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

TPU fixes

Browse Source
This commit is contained in:
ebolam
2022-06-29 17:56:25 -04:00
parent 3e1c9ebacb
commit ce1bff1b84
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
aiserver.py
View File

@@ -4218,7 +4218,7 @@ def tpumtjgenerate(txt, minimum, maximum, found_entries=None):
if(koboldai_vars.lua_koboldbridge.restart_sequence is not None and koboldai_vars.lua_koboldbridge.restart_sequence > 0): if(koboldai_vars.lua_koboldbridge.restart_sequence is not None and koboldai_vars.lua_koboldbridge.restart_sequence > 0):
genresult(genout[koboldai_vars.lua_koboldbridge.restart_sequence-1]["generated_text"]) genresult(genout[koboldai_vars.lua_koboldbridge.restart_sequence-1]["generated_text"])
else: else:
genselect([{"generated_text": x} for x in koboldai_vars.actions.get_current_options()]) genselect([{"generated_text": x['text']} for x in koboldai_vars.actions.get_current_options()])
set_aibusy(0) set_aibusy(0)