mirror of
https://github.com/s0md3v/roop.git
synced 2025-10-16 21:30:37 +08:00
9 lines
203 B
Python
9 lines
203 B
Python
import torch
|
|
import onnxruntime
|
|
|
|
use_gpu = False
|
|
providers = onnxruntime.get_available_providers()
|
|
|
|
if 'TensorrtExecutionProvider' in providers:
|
|
providers.remove('TensorrtExecutionProvider')
|