updating pipeline
Browse filesSigned-off-by: Jacob Renn <jacob.renn@squared.ai>
- 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
|