ZhangYuanhan
commited on
Commit
•
56d9b58
1
Parent(s):
9ac63f9
Update README.md
Browse files
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 |
|