Update README.md
Browse files
README.md
CHANGED
@@ -111,7 +111,7 @@ model_inputs = tokenizer([text], return_tensors="pt").to(device)
|
|
111 |
print(f"Formatted text:\n {text}")
|
112 |
print(f"Model input:\n {model_inputs}")
|
113 |
|
114 |
-
generated_ids = model.generate(model_inputs.input_ids, max_new_tokens=512, do_sample=True, eos_token_id=
|
115 |
generated_ids = [
|
116 |
output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
|
117 |
]
|
|
|
111 |
print(f"Formatted text:\n {text}")
|
112 |
print(f"Model input:\n {model_inputs}")
|
113 |
|
114 |
+
generated_ids = model.generate(model_inputs.input_ids, max_new_tokens=512, do_sample=True, eos_token_id=151645)
|
115 |
generated_ids = [
|
116 |
output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
|
117 |
]
|