nroggendorff commited on
Commit
eab0d43
·
verified ·
1 Parent(s): 269aba9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -21,7 +21,8 @@ def chat(prompt):
21
  max_length=1024,
22
  num_return_sequences=1,
23
  top_p=0.9,
24
- top_k=1,
 
25
  pad_token_id=model.config.eos_token_id
26
  )
27
 
 
21
  max_length=1024,
22
  num_return_sequences=1,
23
  top_p=0.9,
24
+ top_k=50,
25
+ num_beams=2,
26
  pad_token_id=model.config.eos_token_id
27
  )
28