From ce1bff1b8451b60554c4f9e4adb0b7dcfdb1162b Mon Sep 17 00:00:00 2001 From: ebolam Date: Wed, 29 Jun 2022 17:56:25 -0400 Subject: [PATCH] TPU fixes --- aiserver.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aiserver.py b/aiserver.py index 2aa90c2c..a25ae047 100644 --- a/aiserver.py +++ b/aiserver.py @@ -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): genresult(genout[koboldai_vars.lua_koboldbridge.restart_sequence-1]["generated_text"]) 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)