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

Super simple add world info and add world info entry working

Browse Source
This commit is contained in:
ebolam
2022-08-02 21:04:27 -04:00
parent 47c894be8f
commit 55f6a98a3f
3 changed files with 96 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
aiserver.py
View File

@@ -1478,6 +1478,7 @@ def load_model(use_gpu=True, gpu_layers=None, disk_layers=None, initial_load=Fal
global tokenizer
if not utils.HAS_ACCELERATE:
disk_layers = None
koboldai_vars.reset_model()
koboldai_vars.noai = False
if not initial_load:
set_aibusy(True)
@@ -6367,12 +6368,25 @@ def UI_2_Rename_World_Info_Folder(data):
def UI_2_edit_world_info(data):
print("Rename_World_Info_Folder")
print(data)
koboldai_vars.worldinfo_v2.edit_item(data['uid'], data['title'], data['key'],
data['keysecondary'], data['folder'],
data['constant'], data['content'],
data['comment'])
if data['uid'] == -1:
koboldai_vars.worldinfo_v2.add_item(data['title'], data['key'],
data['keysecondary'], data['folder'],
data['constant'], data['content'],
data['comment'])
emit("delete_new_world_info_entry", {})
else:
koboldai_vars.worldinfo_v2.edit_item(data['uid'], data['title'], data['key'],
data['keysecondary'], data['folder'],
data['constant'], data['content'],
data['comment'])
#==================================================================#
# Event triggered when user edits world info item
#==================================================================#
@socketio.on('create_world_info_folder')
def UI_2_create_world_info_folder(data):
koboldai_vars.worldinfo_v2.add_folder("New Folder")
#==================================================================#
# Event triggered to rely a message