Update README.md
Browse files
README.md
CHANGED
@@ -33,7 +33,7 @@ with torch.no_grad():
|
|
33 |
out = model.generate(input_ids,
|
34 |
do_sample=True,
|
35 |
num_beams=4,
|
36 |
-
|
37 |
top_p=0.9,
|
38 |
top_k=50,
|
39 |
max_length=250,
|
|
|
33 |
out = model.generate(input_ids,
|
34 |
do_sample=True,
|
35 |
num_beams=4,
|
36 |
+
# temperature=1.4,
|
37 |
top_p=0.9,
|
38 |
top_k=50,
|
39 |
max_length=250,
|