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

View File

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