apps/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2025-12-24 12:58:02 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fetch available providers

Browse Source
This commit is contained in:
Somdev Sangwan
2023-05-29 17:39:52 +05:30
committed by GitHub
parent c18f66697a
commit ae3b2f8d3c
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
core/processor.py
View File

@@ -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):