Update README.md
Browse files
README.md
CHANGED
|
@@ -34,6 +34,6 @@ messages = [
|
|
| 34 |
|
| 35 |
]
|
| 36 |
inputs = tokenizer.apply_chat_template(messages, return_tensors="pt", add_generation_prompt=True).to(model.device)
|
| 37 |
-
outputs = model.generate(inputs, max_new_tokens=
|
| 38 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
| 39 |
```
|
|
|
|
| 34 |
|
| 35 |
]
|
| 36 |
inputs = tokenizer.apply_chat_template(messages, return_tensors="pt", add_generation_prompt=True).to(model.device)
|
| 37 |
+
outputs = model.generate(inputs, max_new_tokens=256)
|
| 38 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
| 39 |
```
|