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

Saving tweaks on colab

Browse Source
This commit is contained in:
ebolam
2022-09-07 09:09:40 -04:00
parent dcdc00ada4
commit 183e5d9dea
4 changed files with 31 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
aiserver.py
View File

@@ -6771,7 +6771,7 @@ def new_ui_index():
def ui2_connect():
#Send all variables to client
koboldai_vars.send_to_ui()
UI_2_load_tweaks()
pass
#==================================================================#
@@ -7684,6 +7684,22 @@ def UI_2_sp_list_refresh(data):
def UI_2_theme_list_refresh(data):
koboldai_vars.theme_list = [".".join(f.split(".")[:-1]) for f in os.listdir("./themes") if os.path.isfile(os.path.join("./themes", f))]
#==================================================================#
# Save Tweaks
#==================================================================#
@socketio.on('save_tweaks')
def UI_2_save_tweaks(data):
with open("./settings/tweaks.settings", "w") as f:
f.write(data)
#==================================================================#
# Load Tweaks
#==================================================================#
def UI_2_load_tweaks():
if os.path.exists("./settings/tweaks.settings"):
with open("./settings/tweaks.settings", "r") as f:
socketio.emit('load_tweaks', f.read(), room="UI2")
#==================================================================#
# Test
#==================================================================#