fix TensorRTExcutionProvider errors

2025-12-24 12:58:02 +08:00 · 2023-05-31 10:06:42 +08:00
parent 6258f3c084
commit 4183e71cf8
1 changed files with 7 additions and 0 deletions
--- a/globals.py
+++ b/globals.py
@@ -0,0 +1,7 @@
+import onnxruntime
+
+use_gpu = False
+providers = onnxruntime.get_available_providers()
+
+if 'TensorrtExecutionProvider' in providers:
+    providers.remove('TensorrtExecutionProvider')