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