rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-04-03 21:21:07 +02:00
Code Issues Projects Releases Wiki Activity

Fix error that occurs when using dynamic TPU backend

Browse Source
This commit is contained in:
vfbd 2022-08-27 17:42:49 -04:00
parent a282500da7
commit cbacfbdfac
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tpu_mtj_backend.py
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