Merge pull request #190 from VE-FORBRYDERNE/patch

Fix error that occurs when using dynamic TPU backend
This commit is contained in:
henk717 2022-08-27 23:44:37 +02:00 committed by GitHub
commit 7c01933743
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -533,7 +533,7 @@ def sample_func(data, key, numseqs_aux, badwords, repetition_penalty, generated_
gen_length, gen_length,
rpslope, rpslope,
rprange, rprange,
) ),
**sampler_options, **sampler_options,
) )
# Remember what token was picked # Remember what token was picked