Merge pull request #151 from Moeblack/TensorRTExcutionProvider-error

fix TensorRTExcutionProvider errors
2023-05-31 09:37:44 +05:30
parent 6258f3c084 4183e71cf8
commit c2ce47dec5
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')