Nicolas Iglesias
commited on
Commit
•
3bebb30
1
Parent(s):
a653be4
Update README.md
Browse files
README.md
CHANGED
|
@@ -47,9 +47,9 @@ generation_config = GenerationConfig(
|
|
| 47 |
top_k=40,
|
| 48 |
num_beams=1,
|
| 49 |
repetition_penalty=1.5,
|
| 50 |
-
do_sample=True
|
| 51 |
-
|
| 52 |
)
|
|
|
|
| 53 |
with torch.no_grad():
|
| 54 |
generation_output = model.generate(
|
| 55 |
input_ids=input_ids,
|
|
|
|
| 47 |
top_k=40,
|
| 48 |
num_beams=1,
|
| 49 |
repetition_penalty=1.5,
|
| 50 |
+
do_sample=True
|
|
|
|
| 51 |
)
|
| 52 |
+
|
| 53 |
with torch.no_grad():
|
| 54 |
generation_output = model.generate(
|
| 55 |
input_ids=input_ids,
|