jacobrenn commited on
Commit
aadaa02
1 Parent(s): ebe32bd

updating pipeline

Browse files

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

Files changed (1) hide show
  1. instruct_pipeline.py +2 -2
instruct_pipeline.py CHANGED
@@ -8,7 +8,7 @@ INSTRUCTION_KEY = "### Instruction:"
8
  RESPONSE_KEY = "### Response:"
9
  END_KEY = "### End"
10
  INTRO_BLURB = (
11
- "Below is an instruction that describes a task. Write a response that appropriately completes the request."
12
  )
13
 
14
  # This is the prompt that is used for generating responses using an already trained model. It ends with the response
@@ -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
 
8
  RESPONSE_KEY = "### Response:"
9
  END_KEY = "### End"
10
  INTRO_BLURB = (
11
+ "Below is an instruction that describes a task, along with any additional context. Write a response that appropriately completes the request."
12
  )
13
 
14
  # This is the prompt that is used for generating responses using an already trained model. It ends with the response
 
157
  if return_instruction_text:
158
  return {"instruction_text": instruction_text, "generated_text": decoded}
159
 
160
+ return decoded