upperwal commited on
Commit
63d4eb0
·
verified ·
1 Parent(s): c0afea8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -60,10 +60,10 @@ def predict(message, history):
60
  # num_beams=1,
61
  max_new_tokens=300,
62
  do_sample=True,
63
- top_k=5,
64
  num_beams=1,
65
  use_cache=False,
66
- temperature=0.4,
67
  repetition_penalty=1.2,
68
  stopping_criteria=StoppingCriteriaList([stop])
69
  )
 
60
  # num_beams=1,
61
  max_new_tokens=300,
62
  do_sample=True,
63
+ top_k=4,
64
  num_beams=1,
65
  use_cache=False,
66
+ temperature=0.2,
67
  repetition_penalty=1.2,
68
  stopping_criteria=StoppingCriteriaList([stop])
69
  )