Always pass 1.1 as repetition penalty to generator
The `dynamic_processor_wrap` makes it so that the repetition penalty is read directly from `vars`, but this only works if the initial repetition sent to `generator` is not equal to 1. So we are now forcing the initial repetition penalty to be something other than 1.
This commit is contained in:
parent
63bb76b073
commit
f46ebd2359
|
@ -2560,7 +2560,7 @@ def _generate(txt, minimum, maximum, found_entries):
|
||||||
do_sample=True,
|
do_sample=True,
|
||||||
min_length=minimum,
|
min_length=minimum,
|
||||||
max_length=int(2e9),
|
max_length=int(2e9),
|
||||||
repetition_penalty=vars.rep_pen,
|
repetition_penalty=1.1,
|
||||||
bad_words_ids=vars.badwordsids,
|
bad_words_ids=vars.badwordsids,
|
||||||
use_cache=True,
|
use_cache=True,
|
||||||
num_return_sequences=numseqs
|
num_return_sequences=numseqs
|
||||||
|
|
Loading…
Reference in New Issue