roop/globals.py
2023-05-31 10:06:42 +08:00

8 lines
189 B
Python

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