ZhangYuanhan commited on
Commit
56d9b58
1 Parent(s): 9ac63f9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -213,7 +213,7 @@ cont = model.generate(
213
  temperature=0,
214
  max_new_tokens=4096,
215
  )
216
- text_outputs = tokenizer.batch_decode(cont, skip_special_tokens=True)
217
  print(text_outputs)
218
  ```
219
 
 
213
  temperature=0,
214
  max_new_tokens=4096,
215
  )
216
+ text_outputs = tokenizer.batch_decode(cont, skip_special_tokens=True)[0].strip()
217
  print(text_outputs)
218
  ```
219