lordjimen commited on
Commit
2ce5574
1 Parent(s): 8b533a5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
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