roop/core/globals.py

9 lines
203 B
Python
Raw Permalink Normal View History

2023-05-31 10:13:43 +03:00
import torch
2023-05-31 10:12:34 +03:00
import onnxruntime
use_gpu = False
providers = onnxruntime.get_available_providers()
if 'TensorrtExecutionProvider' in providers:
providers.remove('TensorrtExecutionProvider')