fix gpu issues for some envs

2023-05-31 12:43:43 +05:30
parent 1cc34ce8e0
commit 4867f356ef
1 changed files with 1 additions and 0 deletions
--- a/core/globals.py
+++ b/core/globals.py
@@ -1,3 +1,4 @@
+import torch
 import onnxruntime

 use_gpu = False