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

More loading fixes

Browse Source 
My last attempt at fixing this caused GPT2 to break, since the other fix is an edge case we assume that the GPT2 method should be used, and if that fails we try the other one to catch rare errors with bad model config's.
This commit is contained in:
henk717
2022-01-25 06:39:23 +01:00
parent 2bb263c65d
commit 987e78f980
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
aiserver.py
View File

@ -2267,7 +2267,10 @@ def loadmodelsettings():
model_js_config = str(model_config).partition(' ')[2] model_js_config = str(model_config).partition(' ')[2]
js = json.loads(model_js_config) js = json.loads(model_js_config)
except Exception as e: except Exception as e:
model_js_config = open(vars.custmodpth.replace('/', '_') + "/config.json", "r") try:
model_js_config = open(vars.custmodpth + "/config.json", "r")
except Exception as e:
model_js_config = open(vars.custmodpth.replace('/', '_') + "/config.json", "r")
js = json.load(model_js_config) js = json.load(model_js_config)
if("badwordsids" in js): if("badwordsids" in js):
vars.badwordsids = js["badwordsids"] vars.badwordsids = js["badwordsids"]