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