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

Fix for --model with custom paths

Browse Source
This commit is contained in:
ebolam 2022-06-08 18:53:56 -04:00
parent 001439be45
commit 4548dcf1b0
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
aiserver.py
View File

@ -1405,6 +1405,7 @@ def load_model(use_gpu=True, gpu_layers=None, initial_load=False, online_model="
global generator global generator
global torch global torch
global model_config global model_config
global GPT2TokenizerFast
print("Loading vars.model: {} vars.custmodpth: {}".format(vars.model, vars.custmodpth)) print("Loading vars.model: {} vars.custmodpth: {}".format(vars.model, vars.custmodpth))
vars.noai = False vars.noai = False
if not initial_load: if not initial_load: