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
|