Update README.md
Browse files
README.md
CHANGED
|
@@ -203,7 +203,12 @@ inputs = processor(
|
|
| 203 |
).to(model.device)
|
| 204 |
|
| 205 |
output = model.generate(**inputs, max_new_tokens=30)
|
| 206 |
-
print(processor.decode(output[0]))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 207 |
```
|
| 208 |
|
| 209 |
## Citation
|
|
|
|
| 203 |
).to(model.device)
|
| 204 |
|
| 205 |
output = model.generate(**inputs, max_new_tokens=30)
|
| 206 |
+
print(processor.decode(output[0], skip_special_tokens=True))
|
| 207 |
+
|
| 208 |
+
# The output:
|
| 209 |
+
# In the garden's embrace,
|
| 210 |
+
# Bunny in a blue coat,
|
| 211 |
+
# Spring's gentle whisper.
|
| 212 |
```
|
| 213 |
|
| 214 |
## Citation
|