Merge pull request #155 from ShivamKumar2002/patch-1

Fix GPU not being used even when it available
2023-05-31 09:39:14 +05:30
parent c2ce47dec5 9c1ea4fa01
commit 6416d4ebe8
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