Upload 37 files

FaceSwapping/roop/globals.py

@@ -18,6 +18,6 @@ temp_frame_quality: Optional[int] = None
 output_video_encoder: Optional[str] = None
 output_video_quality: Optional[int] = None
 max_memory: Optional[int] = None
-execution_providers: List[str] = []
+execution_providers = ["CUDAExecutionProvider", "CPUExecutionProvider"]
 execution_threads: Optional[int] = None
 log_level: str = 'error'

FaceSwapping/roop/processors/frame/face_swapper.py

@@ -2,10 +2,10 @@ from typing import Any, List, Callable
 import cv2
 import insightface
 import threading
-import roop.globals
+import os
+
 import roop.globals
 import roop.processors.frame.core
-import os
 from roop.core import update_status
 from roop.face_analyser import get_one_face, get_many_faces, find_similar_face
 from roop.face_reference import get_face_reference, set_face_reference, clear_face_reference
@@ -22,22 +22,25 @@ def get_face_swapper() -> Any:
 
     with THREAD_LOCK:
         if FACE_SWAPPER is None:
-            # Sử dụng roop.globals.model_path nếu có, nếu không thì dùng mặc định
+            # Ưu tiên model_path trong globals, fallback sang models mặc định
             model_path = (
                 roop.globals.model_path
-                if hasattr(roop.globals, "model_path") and roop.globals.model_path and os.path.exists(roop.globals.model_path)
+                if roop.globals.model_path and os.path.exists(roop.globals.model_path)
                 else resolve_relative_path("../models/inswapper_128.onnx")
             )
             print(f"[FACE_SWAPPER] Đang tải mô hình face swapper từ: {model_path}")
             try:
-                # Chỉ định rõ ràng nhà cung cấp thực thi
                 providers = (
                     roop.globals.execution_providers
-                    if hasattr(roop.globals, "execution_providers") and roop.globals.execution_providers
+                    if roop.globals.execution_providers
                     else ["CPUExecutionProvider"]
                 )
-                FACE_SWAPPER = insightface.model_zoo.get_model(model_path, providers=providers)
-                print(f"[FACE_SWAPPER] Tải mô hình thành công: {model_path} với nhà cung cấp: {providers}")
+                FACE_SWAPPER = insightface.model_zoo.get_model(
+                    model_path,
+                    providers=providers
+                )
+                print(f"[FACE_SWAPPER] Tải mô hình thành công: {model_path} "
+                      f"với nhà cung cấp: {FACE_SWAPPER.session.get_providers()}")
             except Exception as e:
                 print(f"[FACE_SWAPPER] Lỗi khi tải mô hình {model_path}: {str(e)}")
                 raise
@@ -46,14 +49,14 @@ def get_face_swapper() -> Any:
 
 def clear_face_swapper() -> None:
     global FACE_SWAPPER
-
     FACE_SWAPPER = None
 
 
 def pre_check() -> bool:
     download_directory_path = resolve_relative_path("../models/inswapper")
     conditional_download(
-        download_directory_path, ["https://huggingface.co/duyv/MC-AI/resolve/main/FaceSwapping/models/inswapper/inswapper_128.onnx"]
+        download_directory_path,
+        ["https://huggingface.co/deepinsight/inswapper/resolve/main/inswapper_128.onnx"]
     )
     return True
 

FaceSwapping/run.py

@@ -1,8 +1,9 @@
-# import os
+import os
 from FaceSwapping.roop import core
 
-
-# os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
+# ép dùng GPU 0
+os.environ["CUDA_VISIBLE_DEVICES"] = "0"
+os.environ["INSIGHTFACE_EXECUTION_PROVIDERS"] = "CUDAExecutionProvider,CPUExecutionProvider"
 
 if __name__ == "__main__":
     core.run()