From ae3b2f8d3c9324139733dbbb3e27fbf9325e4a05 Mon Sep 17 00:00:00 2001 From: Somdev Sangwan Date: Mon, 29 May 2023 17:39:52 +0530 Subject: [PATCH] fetch available providers --- core/processor.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/core/processor.py b/core/processor.py index a7c5265..535cc17 100644 --- a/core/processor.py +++ b/core/processor.py @@ -1,9 +1,10 @@ import cv2 import insightface +import onnxruntime from core.config import get_face from core.utils import rreplace -face_swapper = insightface.model_zoo.get_model('inswapper_128.onnx', providers=['CUDAExecutionProvider', 'ROCMExecutionProvider', 'CPUExecutionProvider']) +face_swapper = insightface.model_zoo.get_model('inswapper_128.onnx', providers=onnxruntime.get_available_providers()) def process_video(source_img, frame_paths):