Update README.md
Browse files
README.md
CHANGED
|
@@ -102,10 +102,10 @@ with torch.no_grad():
|
|
| 102 |
outputs = model.generate(
|
| 103 |
**inputs,
|
| 104 |
max_new_tokens=200,
|
| 105 |
-
temperature=0.
|
| 106 |
top_p=0.9,
|
| 107 |
do_sample=True,
|
| 108 |
-
repetition_penalty=1.
|
| 109 |
)
|
| 110 |
|
| 111 |
print(tokenizer.decode(outputs[0], skip_special_tokens=False))
|
|
|
|
| 102 |
outputs = model.generate(
|
| 103 |
**inputs,
|
| 104 |
max_new_tokens=200,
|
| 105 |
+
temperature=0.3,
|
| 106 |
top_p=0.9,
|
| 107 |
do_sample=True,
|
| 108 |
+
repetition_penalty=1.16,
|
| 109 |
)
|
| 110 |
|
| 111 |
print(tokenizer.decode(outputs[0], skip_special_tokens=False))
|