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

Update app.py

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