Merge pull request #38 from j129008/main

load model function fix
2023-06-22 05:30:33 +05:30 · 2023-06-22 05:30:33 +05:30 · 9cf85b19f8
parent 4661a37d05 0a14e53f47
commit 9cf85b19f8
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