diff --git a/setup.py b/setup.py
index 46abccac..67bf114a 100644
--- a/setup.py
+++ b/setup.py
@@ -38,7 +38,6 @@
     "safetensors",
     "torch>=2.6",
     "torchprofile>=0.0.4",
-    "torchvision",
 ]
 
 optional_deps = {
@@ -79,6 +78,7 @@
         "pytest-cov",
         "pytest-timeout",
         "timm",
+        "torchvision",
         "tox>4.18",
         "tox-current-env>=0.0.12",
     ],