updating pipeline
Browse filesSigned-off-by: Jacob Renn <jacob.renn@squared.ai>
- instruct_pipeline.py +1 -1
instruct_pipeline.py
CHANGED
@@ -157,4 +157,4 @@ class InstructionTextGenerationPipeline(Pipeline):
|
|
157 |
if return_instruction_text:
|
158 |
return {"instruction_text": instruction_text, "generated_text": decoded}
|
159 |
|
160 |
-
return decoded
|
|
|
157 |
if return_instruction_text:
|
158 |
return {"instruction_text": instruction_text, "generated_text": decoded}
|
159 |
|
160 |
+
return {'generated_text': decoded}
|