load model function fix

2023-06-22 06:03:36 +08:00 · 2023-06-22 06:03:36 +08:00 · 29acad245e
parent 4ce41b6b34
commit 29acad245e
1 changed files with 1 additions and 1 deletions
--- a/scripts/faceswap.py
+++ b/scripts/faceswap.py
@ -18,8 +18,8 @@ import os


 def get_models():
-    models = glob.glob(models_path)
    models_path = os.path.join(scripts.basedir(), "models/roop/*")
+    models = glob.glob(models_path)
    models += glob.glob(models_path)
    models = [x for x in models if x.endswith(".onnx") or x.endswith(".pth")]
    return models