Update README.md
Browse files
README.md
CHANGED
|
@@ -127,8 +127,8 @@ input_ids = tokenizer(prompt, return_tensors="pt", padding=True, max_length=40,
|
|
| 127 |
|
| 128 |
# Translation
|
| 129 |
with torch.no_grad():
|
| 130 |
-
generated_ids = model.generate(input_ids=input_ids, num_beams=5, max_new_tokens=20, do_sample=True, temperature=0.6, top_p=0.9)
|
| 131 |
-
outputs = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
|
| 132 |
print(outputs)
|
| 133 |
```
|
| 134 |
|
|
|
|
| 127 |
|
| 128 |
# Translation
|
| 129 |
with torch.no_grad():
|
| 130 |
+
generated_ids = model.generate(input_ids=input_ids, num_beams=5, max_new_tokens=20, do_sample=True, temperature=0.6, top_p=0.9)
|
| 131 |
+
outputs = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
|
| 132 |
print(outputs)
|
| 133 |
```
|
| 134 |
|