Skip to content

Commit 79abf8d

Browse files
authored
[Text Generation Pipeline] Fix Non KV Cache Pipeline not calling parse_inputs (#1552)
* Update pipeline_no_kv_cache.py 8) * grrr
1 parent a5003e3 commit 79abf8d

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

src/deepsparse/transformers/pipelines/text_generation/pipeline_no_kv_cache.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -101,9 +101,7 @@ def __init__(
101101

102102
# TODO: Using the GraphRouter, but should use
103103
# LinearRouter with appropriate split/join support
104-
router = GraphRouter(
105-
end_route="STOP", start_route="process_input", route=routes
106-
)
104+
router = GraphRouter(end_route="STOP", start_route="parse_inputs", route=routes)
107105
scheduler = [OperatorScheduler()]
108106
super().__init__(
109107
ops=ops,

0 commit comments

Comments
 (0)