Files
roop/core/globals.py
Somdev Sangwan 1cc34ce8e0 bruh moment 12
2023-05-31 12:42:34 +05:30

8 lines
189 B
Python

import onnxruntime
use_gpu = False
providers = onnxruntime.get_available_providers()
if 'TensorrtExecutionProvider' in providers:
providers.remove('TensorrtExecutionProvider')