rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-02-02 02:26:45 +01:00
Code Issues Projects Releases Wiki Activity

Update aiserver.py

Browse Source 
Better way of checking for the model type
This commit is contained in:
henk717 2021-09-25 16:50:24 +02:00
parent 7d35f825c6
commit c9290d02dc
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
aiserver.py
View File

@ -404,7 +404,7 @@ if(not vars.model in ["InferKit", "Colab", "OAI", "ReadOnly"]):
if(vars.model == "NeoCustom"): if(vars.model == "NeoCustom"):
model_config = open(vars.custmodpth + "/config.json", "r") model_config = open(vars.custmodpth + "/config.json", "r")
js = json.load(model_config) js = json.load(model_config)
if("architectures" in js): if("model_type" in js):
model = vars.custmodpth model = vars.custmodpth
else: else:
model = GPTNeoForCausalLM.from_pretrained(vars.custmodpth) model = GPTNeoForCausalLM.from_pretrained(vars.custmodpth)