Luigi commited on
Commit
ab3ef88
·
1 Parent(s): 3b87f8e

fix for ValueError: `diversity_penalty` should be a float strictly larger than 0.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -81,7 +81,7 @@ def suggest_next(text, model_name, k, m, num_beam_groups, diversity_penalty):
81
  }
82
  if diversity_penalty and diversity_penalty > 0:
83
  gen_kwargs["num_beam_groups"] = num_beam_groups
84
- gen_kwargs["diversity_penalty"] = diversity_penalty
85
 
86
  outs = gen_pipe(text, **gen_kwargs)
87
 
 
81
  }
82
  if diversity_penalty and diversity_penalty > 0:
83
  gen_kwargs["num_beam_groups"] = num_beam_groups
84
+ gen_kwargs["diversity_penalty"] = float(diversity_penalty)
85
 
86
  outs = gen_pipe(text, **gen_kwargs)
87