updating model and pipeline
Browse filesSigned-off-by: Jacob Renn <jacob.renn@squared.ai>
- instruct_pipeline.py +2 -2
- pytorch_model.bin +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
|
pytorch_model.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1bff91718ec1dd0384f2f1c3ef43d451f8a96c53fab163d9ae0fcb426c924f0f
|
| 3 |
+
size 734924381
|