lordjimen commited on
Commit
ce97d72
·
verified ·
1 Parent(s): 2a2f4e3

Update README.md

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