Fix for --model with custom paths
This commit is contained in:
parent
001439be45
commit
4548dcf1b0
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue