saurabh5 commited on
Commit
53d899c
·
verified ·
1 Parent(s): b9cbb32

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -179,7 +179,7 @@ model = AutoModelForCausalLM.from_pretrained(
179
  device_map="auto",
180
  )
181
 
182
- prompt = "Who would in in a fight - a dinosaur or a cow named MooMoo?"
183
  inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
184
 
185
  outputs = model.generate(
@@ -205,7 +205,7 @@ sampling_params = SamplingParams(
205
  max_tokens=32768,
206
  )
207
 
208
- prompt = "Who would in in a fight - a dinosaur or a cow named MooMoo?"
209
  outputs = llm.generate(prompt, sampling_params)
210
  print(outputs[0].outputs[0].text)
211
  ```
 
179
  device_map="auto",
180
  )
181
 
182
+ prompt = "Who would win in a fight - a dinosaur or a cow named MooMoo?"
183
  inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
184
 
185
  outputs = model.generate(
 
205
  max_tokens=32768,
206
  )
207
 
208
+ prompt = "Who would win in a fight - a dinosaur or a cow named MooMoo?"
209
  outputs = llm.generate(prompt, sampling_params)
210
  print(outputs[0].outputs[0].text)
211
  ```