CreitinGameplays commited on
Commit
b3b3337
1 Parent(s): 8e23326

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ def generate(
59
  top_p=top_p,
60
  top_k=top_k,
61
  temperature=temperature,
62
- num_beams=1,
63
  repetition_penalty=repetition_penalty,
64
  )
65
  t = Thread(target=model.generate, kwargs=generate_kwargs)
 
59
  top_p=top_p,
60
  top_k=top_k,
61
  temperature=temperature,
62
+ num_beams=5,
63
  repetition_penalty=repetition_penalty,
64
  )
65
  t = Thread(target=model.generate, kwargs=generate_kwargs)