habdine commited on
Commit
c9e74a0
·
verified ·
1 Parent(s): ff2fa8a

Update app.py

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