Brasd99 commited on
Commit
1e318ff
1 Parent(s): b2e46de
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ def generate(
28
  model, tok, text,
29
  do_sample=True, max_length=10000, repetition_penalty=5.0,
30
  top_k=5, top_p=0.95, temperature=1,
31
- num_beams=None,
32
  no_repeat_ngram_size=3
33
  ):
34
  input_ids = tok.encode(text, return_tensors="pt").to(DEVICE)
 
28
  model, tok, text,
29
  do_sample=True, max_length=10000, repetition_penalty=5.0,
30
  top_k=5, top_p=0.95, temperature=1,
31
+ num_beams=10,
32
  no_repeat_ngram_size=3
33
  ):
34
  input_ids = tok.encode(text, return_tensors="pt").to(DEVICE)