diff --git a/modules/models.py b/modules/models.py index 062ccb1f..a2256b98 100644 --- a/modules/models.py +++ b/modules/models.py @@ -109,6 +109,7 @@ def load_model(model_name): # check root of models folder, and model path root paths = [ f"{path_to_model}/{pt_model}", f"models/{pt_model}" ] + pt_path = None for path in [ Path(p) for p in paths ]: if path.exists(): pt_path = path