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

bruh moment 12

Browse Source
This commit is contained in:
Somdev Sangwan 2023-05-31 12:42:34 +05:30
parent 077b931cef
commit 1cc34ce8e0
2 changed files with 7 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
core.globals.py
View File

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

12
core/globals.py
View File

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