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

fix TensorRTExcutionProvider errors

Browse Source
This commit is contained in:
Pikachu~~~ 2023-05-31 10:06:42 +08:00 committed by GitHub
parent 6258f3c084
commit 4183e71cf8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
globals.py Normal file
View File

@ -0,0 +1,7 @@
import onnxruntime
use_gpu = False
providers = onnxruntime.get_available_providers()
if 'TensorrtExecutionProvider' in providers:
providers.remove('TensorrtExecutionProvider')