sidpanda commited on
Commit
62d041b
·
verified ·
1 Parent(s): 0111d3f

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=1,
65
  use_cache=False,
66
- temperature=0.5,
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=3,
64
+ num_beams=4,
65
  use_cache=False,
66
+ temperature=0.2,
67
  repetition_penalty=1.2,
68
  stopping_criteria=StoppingCriteriaList([stop])
69
  )