adj

.gitignore

@@ -0,0 +1,12 @@
+output/
+roop/__pycache__
+roop/processors/frame/__pycache__
+roop/frame/__pycache__
+flagged/
+*.jpg
+*.pyc
+*.pth
+*.onnx
+.idea
+.vscode
+models/

app.py

@@ -46,7 +46,7 @@ def swap_face(source_file, target_file,doFaceEnhancer):
     roop.globals.video_encoder = "libx264"
     roop.globals.video_quality = 18
     roop.globals.max_memory = suggest_max_memory()
-    roop.globals.execution_providers = decode_execution_providers(["cpu"])
+    roop.globals.execution_providers = decode_execution_providers(["cuda"])
     roop.globals.execution_threads = suggest_execution_threads()
 
     print(

roop/processors/frame/face_enhancer.py

@@ -9,6 +9,7 @@ from roop.core import update_status
 from roop.face_analyser import get_one_face
 from roop.typing import Frame, Face
 from roop.utilities import conditional_download, resolve_relative_path, is_image, is_video
+import torch
 
 FACE_ENHANCER = None
 THREAD_SEMAPHORE = threading.Semaphore()
@@ -16,6 +17,11 @@ THREAD_LOCK = threading.Lock()
 NAME = 'ROOP.FACE-ENHANCER'
 frame_name = 'face_enhancer'
 
+if torch.cuda.is_available():
+    device='cuda'
+else:
+    device='cpu'
+
 
 def get_face_enhancer() -> Any:
     global FACE_ENHANCER
@@ -24,7 +30,7 @@ def get_face_enhancer() -> Any:
         if FACE_ENHANCER is None:
             model_path = resolve_relative_path('../models/GFPGANv1.4.pth')
             # todo: set models path https://github.com/TencentARC/GFPGAN/issues/399
-            FACE_ENHANCER = gfpgan.GFPGANer(model_path=model_path, upscale=1) # type: ignore[attr-defined]
+            FACE_ENHANCER = gfpgan.GFPGANer(model_path=model_path, upscale=1,device=device) # type: ignore[attr-defined]
     return FACE_ENHANCER