Fix for --model with custom paths

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

View File

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