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 disconnect on story load for large stories

Browse Source
This commit is contained in:
ebolam
2022-09-29 13:49:34 -04:00
parent ff6183c419
commit 4c97c291f1
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
koboldai_settings.py
View File

@@ -6,6 +6,7 @@ from flask_socketio import SocketIO
from collections import OrderedDict from collections import OrderedDict
import multiprocessing import multiprocessing
from logger import logger from logger import logger
import eventlet
serverstarted = False serverstarted = False
queue = None queue = None
@@ -1330,6 +1331,7 @@ class KoboldStoryRegister(object):
if self.actions[key]['In AI Input']: if self.actions[key]['In AI Input']:
self.actions[key]['Selected Text Length'] = len(self.tokenizer.encode(self.actions[key]['Selected Text'])) self.actions[key]['Selected Text Length'] = len(self.tokenizer.encode(self.actions[key]['Selected Text']))
process_variable_changes(self.socketio, "story", 'actions', {"id": key, 'action': self.actions[key]}, None) process_variable_changes(self.socketio, "story", 'actions', {"id": key, 'action': self.actions[key]}, None)
eventlet.sleep(0.01)
else: else:
for key in self.actions: for key in self.actions:
self.actions[key]['Selected Text Length'] = 0 self.actions[key]['Selected Text Length'] = 0