upperwal commited on
Commit
df889e0
·
verified ·
1 Parent(s): 57c494d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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=4,
64
- num_beams=4,
65
  use_cache=False,
66
- temperature=0.2,
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=5,
64
+ num_beams=1,
65
  use_cache=False,
66
+ temperature=0.4,
67
  repetition_penalty=1.2,
68
  stopping_criteria=StoppingCriteriaList([stop])
69
  )