apparently torchvision is required

TNTwise · Apr 24, 2024 · 7d1052d · 7d1052d
1 parent ec09b1d
commit 7d1052d
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/main.py b/main.py
@@ -31,6 +31,7 @@
 
 try:
     import torch
+    import torchvision
     import spandrel
 
     torch_version = True

diff --git a/src/getModels/select_models.py b/src/getModels/select_models.py
@@ -53,7 +53,8 @@ def handleCUDAModels(model: str = ""):
 
 try:
     import torch
-
+    import torchvision
+    import spandrel
     torch_version = True
 
 except Exception as e:

diff --git a/src/misc/onProgramStart.py b/src/misc/onProgramStart.py
@@ -17,6 +17,7 @@
     gmfss = False
 try:
     import torch
+    import torchvision
     import spandrel
 
     torch_version = True