bump torchvision

neuralmagic · Jun 9, 2023 · d428930 · d428930
1 parent b82fb6f
commit d428930
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -68,11 +68,11 @@
     "gputils",
 ]
 _pytorch_all_deps = _pytorch_deps + [
-    "torchvision>=0.3.0,<=0.15",
+    "torchvision>=0.3.0,<=0.15.1",
     "torchaudio<=0.13",
 ]
 _pytorch_vision_deps = _pytorch_deps + [
-    "torchvision>=0.3.0,<=0.15",
+    "torchvision>=0.3.0,<=0.15.1",
     "opencv-python<=4.6.0.66",
 ]
 _transformers_deps = _pytorch_deps + [