RuVI-AI
/

ArushCodes commited on
Commit
ce87a82
·
verified ·
1 Parent(s): f766633

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -191,10 +191,11 @@ out = model.generate(
191
  max_new_tokens=60,
192
  do_sample=True,
193
  temperature=0.7,
194
- top_k=20,
195
  repetition_penalty=1.3,
196
  no_repeat_ngram_size=3,
197
  pad_token_id=tok.pad_token_id,
 
198
  )
199
  print(tok.decode(out[0], skip_special_tokens=True))
200
  ```
@@ -219,9 +220,10 @@ These settings gave fluent, on-topic output in testing:
219
 
220
  ```text
221
  temperature 0.7
222
- top_k 20
223
  repetition_penalty 1.3
224
  no_repeat_ngram 3
 
225
  ```
226
 
227
  Greedy decoding (`do_sample=False`) repeats phrases after a few sentences, so use sampling.
 
191
  max_new_tokens=60,
192
  do_sample=True,
193
  temperature=0.7,
194
+ top_k=40,
195
  repetition_penalty=1.3,
196
  no_repeat_ngram_size=3,
197
  pad_token_id=tok.pad_token_id,
198
+ top_p=0.95
199
  )
200
  print(tok.decode(out[0], skip_special_tokens=True))
201
  ```
 
220
 
221
  ```text
222
  temperature 0.7
223
+ top_k 40
224
  repetition_penalty 1.3
225
  no_repeat_ngram 3
226
+ top_p 0.95
227
  ```
228
 
229
  Greedy decoding (`do_sample=False`) repeats phrases after a few sentences, so use sampling.