jacobrenn commited on
Commit
dc8f7f5
1 Parent(s): 3175c43

updating model and pipeline

Browse files

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

Files changed (2) hide show
  1. instruct_pipeline.py +2 -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:4fc8625a3d1c95eed89421c80c23f3a49c7c0937306aafcc4b15a27f4e3277dd
3
- size 1444581661
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1bff91718ec1dd0384f2f1c3ef43d451f8a96c53fab163d9ae0fcb426c924f0f
3
+ size 734924381