bruh moment 12

2023-05-31 12:42:34 +05:30
parent 077b931cef
commit 1cc34ce8e0
2 changed files with 7 additions and 12 deletions
--- a/core.globals.py
+++ b/core.globals.py
@@ -1,7 +0,0 @@
-import onnxruntime
-
-use_gpu = False
-providers = onnxruntime.get_available_providers()
-
-if 'TensorrtExecutionProvider' in providers:
-    providers.remove('TensorrtExecutionProvider')
--- a/core/globals.py
+++ b/core/globals.py
@@ -1,5 +1,7 @@
-import torch
 import onnxruntime

 use_gpu = False
 providers = onnxruntime.get_available_providers()
+
+if 'TensorrtExecutionProvider' in providers:
+    providers.remove('TensorrtExecutionProvider')