From 739a8a5268c0ae08b296925d29d65a36cc80c0a5 Mon Sep 17 00:00:00 2001 From: KoboldAI Dev Date: Mon, 10 May 2021 09:26:31 -0400 Subject: [PATCH] Bugfix: Check for keys in client.settings before attempting to access --- aiserver.py | 24 ++++++++++++++---------- 1 file changed, 14 insertions(+), 10 deletions(-) diff --git a/aiserver.py b/aiserver.py index 95c192f5..fecb6f43 100644 --- a/aiserver.py +++ b/aiserver.py @@ -401,16 +401,20 @@ def loadsettings(): js = json.load(file) # Copy file contents to vars - #for set in js: - # vars[set] = js[set] - vars.apikey = js["apikey"] - vars.andepth = js["andepth"] - vars.temp = js["temp"] - vars.top_p = js["top_p"] - vars.rep_pen = js["rep_pen"] - vars.genamt = js["genamt"] - vars.max_length = js["max_length"] - vars.ikgen = js["ikgen"] + if("andepth" in js): + vars.andepth = js["andepth"] + if("temp" in js): + vars.temp = js["temp"] + if("top_p" in js): + vars.top_p = js["top_p"] + if("rep_pen" in js): + vars.rep_pen = js["rep_pen"] + if("genamt" in js): + vars.genamt = js["genamt"] + if("max_length" in js): + vars.max_length = js["max_length"] + if("ikgen" in js): + vars.ikgen = js["ikgen"] file.close()