upperwal commited on
Commit
808f932
·
verified ·
1 Parent(s): b3d9c60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -64,6 +64,7 @@ def predict(message, history):
64
  num_beams=1,
65
  use_cache=False,
66
  temperature=0.4,
 
67
  stopping_criteria=StoppingCriteriaList([stop])
68
  )
69
  t = Thread(target=model.generate, kwargs=generate_kwargs)
 
64
  num_beams=1,
65
  use_cache=False,
66
  temperature=0.4,
67
+ repetition_penalty=10.,
68
  stopping_criteria=StoppingCriteriaList([stop])
69
  )
70
  t = Thread(target=model.generate, kwargs=generate_kwargs)