saved layer value fix

2022-03-07 19:21:55 -05:00 · 2022-03-07 19:21:55 -05:00 · d8ab58892d
parent da53d7edb3
commit d8ab58892d
1 changed files with 2 additions and 0 deletions
--- a/aiserver.py
+++ b/aiserver.py
@ -2570,6 +2570,8 @@ def get_message(msg):
    elif(msg['cmd'] == 'list_model'):
        sendModelSelection(menu=msg['data'])
    elif(msg['cmd'] == 'load_model'):
+        if not os.path.exists("settings/" + vars.model.replace('/', '_') + ".breakmodel"):
+            os.mkdir("settings")
        f = open("settings/" + vars.model.replace('/', '_') + ".breakmodel", "w")
        f.write(msg['gpu_layers'])
        f.close()