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
|