Update README.md
Browse files
README.md
CHANGED
@@ -94,7 +94,8 @@ generation_params = GenerationConfig(
|
|
94 |
top_k=25,
|
95 |
top_p=1,
|
96 |
repetition_penalty=1.1,
|
97 |
-
eos_token_id=[1,107]
|
|
|
98 |
)
|
99 |
```
|
100 |
|
|
|
94 |
top_k=25,
|
95 |
top_p=1,
|
96 |
repetition_penalty=1.1,
|
97 |
+
eos_token_id=[1,107],
|
98 |
+
do_sample=True
|
99 |
)
|
100 |
```
|
101 |
|