doberst commited on
Commit
c2902b7
1 Parent(s): 92e2f50

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +23 -3
README.md CHANGED
@@ -97,9 +97,29 @@ To get the best results, package "my_prompt" as follows:
97
  my_prompt = {{text_passage}} + "\n" + {{question/instruction}}
98
 
99
 
 
100
 
101
- ## Model Card Contact
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
102
 
103
- Darren Oberst & llmware team
 
 
 
104
 
105
- Please reach out anytime if you are interested in this project!
 
97
  my_prompt = {{text_passage}} + "\n" + {{question/instruction}}
98
 
99
 
100
+ If you are using a HuggingFace generation script:
101
 
102
+ # prepare prompt packaging used in fine-tuning process
103
+ new_prompt = "<human>: " + entries["context"] + "\n" + entries["query"] + "\n" + "<bot>:"
104
+
105
+ inputs = tokenizer(new_prompt, return_tensors="pt")
106
+ start_of_output = len(inputs.input_ids[0])
107
+
108
+ # temperature: set at 0.3 for consistency of output
109
+ # max_new_tokens: set at 100 - may prematurely stop a few of the summaries
110
+
111
+ outputs = model.generate(
112
+ inputs.input_ids.to(device),
113
+ eos_token_id=tokenizer.eos_token_id,
114
+ pad_token_id=tokenizer.eos_token_id,
115
+ do_sample=True,
116
+ temperature=0.3,
117
+ max_new_tokens=100,
118
+ )
119
 
120
+ output_only = tokenizer.decode(outputs[0][start_of_output:],skip_special_tokens=True)
121
+
122
+
123
+ ## Model Card Contact
124
 
125
+ Darren Oberst & llmware team