Update README.md
Browse files
README.md
CHANGED
|
@@ -121,7 +121,7 @@ text = tokenizer.apply_chat_template(
|
|
| 121 |
add_generation_prompt=True,
|
| 122 |
enable_thinking=True
|
| 123 |
)
|
| 124 |
-
model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
|
| 125 |
|
| 126 |
generated_ids = model.generate(
|
| 127 |
**model_inputs,
|
|
|
|
| 121 |
add_generation_prompt=True,
|
| 122 |
enable_thinking=True
|
| 123 |
)
|
| 124 |
+
model_inputs = tokenizer([text], return_tensors="pt", return_token_type_ids=False).to(model.device)
|
| 125 |
|
| 126 |
generated_ids = model.generate(
|
| 127 |
**model_inputs,
|