Update README.md
Browse files
README.md
CHANGED
|
@@ -84,7 +84,8 @@ generation_params = GenerationConfig(
|
|
| 84 |
top_k=25,
|
| 85 |
top_p=1,
|
| 86 |
repetition_penalty=1.1,
|
| 87 |
-
eos_token_id=[1,107]
|
|
|
|
| 88 |
)
|
| 89 |
```
|
| 90 |
|
|
|
|
| 84 |
top_k=25,
|
| 85 |
top_p=1,
|
| 86 |
repetition_penalty=1.1,
|
| 87 |
+
eos_token_id=[1,107],
|
| 88 |
+
do_sample=True
|
| 89 |
)
|
| 90 |
```
|
| 91 |
|