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 |
|