aixsatoshi commited on
Commit
65ac526
1 Parent(s): 764ca7f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -92,6 +92,7 @@ def predict(message, history, tokens, temperature, language):
92
  do_sample=True,
93
  top_p=0.95,
94
  top_k=20,
 
95
  num_beams=1,
96
  stopping_criteria=StoppingCriteriaList([stop])
97
  )
@@ -198,6 +199,7 @@ def predict(message, tokens, temperature, language):
198
  do_sample=True,
199
  top_p=0.95,
200
  top_k=20,
 
201
  num_beams=1,
202
  eos_token_id=tokenizer.eos_token_id
203
  )
 
92
  do_sample=True,
93
  top_p=0.95,
94
  top_k=20,
95
+ repetition_penalty=1.15,
96
  num_beams=1,
97
  stopping_criteria=StoppingCriteriaList([stop])
98
  )
 
199
  do_sample=True,
200
  top_p=0.95,
201
  top_k=20,
202
+ repetition_penalty=1.15,
203
  num_beams=1,
204
  eos_token_id=tokenizer.eos_token_id
205
  )