fix reference issue
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user