upagge/roop
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fetch available providers

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

3
core/config.py
View File

@@ -1,6 +1,7 @@
import insightface
import onnxruntime
face_analyser = insightface.app.FaceAnalysis(name='buffalo_l', providers=['CUDAExecutionProvider', 'ROCMExecutionProvider', 'CPUExecutionProvider'])
face_analyser = insightface.app.FaceAnalysis(name='buffalo_l', providers=onnxruntime.get_available_providers())
face_analyser.prepare(ctx_id=0, det_size=(640, 640))