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

Optimization

Browse Source
This commit is contained in:
ebolam
2022-10-01 16:12:59 -04:00
parent 928ec5c411
commit 5b68c2d64d
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
koboldai_settings.py
View File

@@ -1510,15 +1510,17 @@ class KoboldWorldInfo(object):
if uid is not None: if uid is not None:
if uid in self.world_info: if uid in self.world_info:
self.world_info[uid]['token_length'] = len(self.tokenizer.encode(self.world_info[uid]['content'])) self.world_info[uid]['token_length'] = len(self.tokenizer.encode(self.world_info[uid]['content']))
self.socketio.emit("world_info_entry", self.world_info[uid], broadcast=True, room="UI_2")
else: else:
for uid in self.world_info: for uid in self.world_info:
self.world_info[uid]['token_length'] = len(self.tokenizer.encode(self.world_info[uid]['content'])) self.world_info[uid]['token_length'] = len(self.tokenizer.encode(self.world_info[uid]['content']))
self.send_to_ui()
else: else:
for uid in self.world_info: for uid in self.world_info:
self.world_info[uid]['token_length'] = 0 self.world_info[uid]['token_length'] = 0
self.send_to_ui() self.send_to_ui()
def add_folder(self, folder): def add_folder(self, folder):
if folder in self.world_info_folder: if folder in self.world_info_folder:
i=0 i=0