diff --git a/globals.py b/globals.py new file mode 100644 index 0000000..3525c9d --- /dev/null +++ b/globals.py @@ -0,0 +1,7 @@ +import onnxruntime + +use_gpu = False +providers = onnxruntime.get_available_providers() + +if 'TensorrtExecutionProvider' in providers: + providers.remove('TensorrtExecutionProvider')