From a116b2cb718ecc8f4b90a606113021dce27eaefc Mon Sep 17 00:00:00 2001 From: Somdev Sangwan Date: Thu, 22 Jun 2023 16:17:58 +0530 Subject: [PATCH] enable GPU support --- scripts/swapper.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/scripts/swapper.py b/scripts/swapper.py index d86f4f2..76e5df2 100644 --- a/scripts/swapper.py +++ b/scripts/swapper.py @@ -17,7 +17,7 @@ from modules.face_restoration import FaceRestoration, restore_faces from modules.upscaler import Upscaler, UpscalerData from scripts.roop_logging import logger -providers = ["CPUExecutionProvider"] +providers = onnxruntime.get_available_providers() @dataclass @@ -28,8 +28,6 @@ class UpscaleOptions: face_restorer: FaceRestoration = None restorer_visibility: float = 0.5 -ANALYSIS_MODEL = insightface.app.FaceAnalysis(name="buffalo_l", providers=providers) - FS_MODEL = None CURRENT_FS_MODEL_PATH = None @@ -75,7 +73,7 @@ def upscale_image(image: Image, upscale_options: UpscaleOptions): def get_face_single(img_data: np.ndarray, face_index=0, det_size=(640, 640)): - face_analyser = copy.deepcopy(ANALYSIS_MODEL) + face_analyser = insightface.app.FaceAnalysis(name="buffalo_l", providers=providers) face_analyser.prepare(ctx_id=0, det_size=det_size) face = face_analyser.get(img_data)