pszemraj commited on
Commit
5db51ab
1 Parent(s): c263ed8

num_beam_groups to 1 default

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def generate_text(
20
  num_beams=4,
21
  no_repeat_ngram_size=2,
22
  length_penalty=1.0,
23
- num_beam_groups=2,
24
  # perma params (not set by user)
25
  repetition_penalty=3.5,
26
  abs_max_length=512,
20
  num_beams=4,
21
  no_repeat_ngram_size=2,
22
  length_penalty=1.0,
23
+ num_beam_groups=1,
24
  # perma params (not set by user)
25
  repetition_penalty=3.5,
26
  abs_max_length=512,