Update README.md
Browse files
README.md
CHANGED
|
@@ -59,7 +59,7 @@ tokenized_chat = tokenizer.apply_chat_template(
|
|
| 59 |
generated_ids = model.generate(tokenized_chat, max_new_tokens=512)
|
| 60 |
|
| 61 |
generated_ids = [
|
| 62 |
-
output_ids[len(input_ids):] for input_ids, output_ids in zip(
|
| 63 |
]
|
| 64 |
|
| 65 |
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
|
|
|
| 59 |
generated_ids = model.generate(tokenized_chat, max_new_tokens=512)
|
| 60 |
|
| 61 |
generated_ids = [
|
| 62 |
+
output_ids[len(input_ids):] for input_ids, output_ids in zip(tokenized_chat, generated_ids)
|
| 63 |
]
|
| 64 |
|
| 65 |
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
|