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

Fix some of the logic for generation aborting

Browse Source
This commit is contained in:
Gnome Ann
2022-01-10 17:09:47 -05:00
parent 902b6b0cee
commit 43586c8f60
3 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
aiserver.py
View File

@ -1708,8 +1708,9 @@ def get_message(msg):
# Submit action
if(msg['cmd'] == 'submit'):
if(vars.mode == "play"):
if(vars.aibusy and msg.get('allowabort', False)):
vars.abort = True
if(vars.aibusy):
if(msg.get('allowabort', False)):
vars.abort = True
return
vars.abort = False
vars.lua_koboldbridge.feedback = None
@ -1727,8 +1728,9 @@ def get_message(msg):
memsubmit(msg['data'])
# Retry Action
elif(msg['cmd'] == 'retry'):
if(vars.aibusy and msg.get('allowabort', False)):
vars.abort = True
if(vars.aibusy):
if(msg.get('allowabort', False)):
vars.abort = True
return
vars.abort = False
if(vars.chatmode):