jacobrenn commited on
Commit
f98c8d1
1 Parent(s): 3542e80

updating pipeline

Browse files

Signed-off-by: Jacob Renn <jacob.renn@squared.ai>

Files changed (1) hide show
  1. 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}