Update README.md
Browse files
README.md
CHANGED
|
@@ -73,6 +73,7 @@ model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
|
|
| 73 |
|
| 74 |
generated_ids = model.generate(
|
| 75 |
model_inputs["input_ids"],
|
|
|
|
| 76 |
max_new_tokens=512
|
| 77 |
)
|
| 78 |
generated_ids = [
|
|
|
|
| 73 |
|
| 74 |
generated_ids = model.generate(
|
| 75 |
model_inputs["input_ids"],
|
| 76 |
+
attention_mask=model_inputs["attention_mask"],
|
| 77 |
max_new_tokens=512
|
| 78 |
)
|
| 79 |
generated_ids = [
|