allow patch versions of torchvision 0.14 beyond .0 (#1556) (#1557)

as written, only 0.14.0 would be allowed
neuralmagic · May 10, 2023 · b048ad7 · b048ad7
1 parent 1c2ee2c
commit b048ad7
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -66,11 +66,10 @@
     "gputils",
 ]
 _pytorch_all_deps = _pytorch_deps + [
-    "torchvision>=0.3.0,<=0.14",
+    "torchvision>=0.3.0,<0.15",
     "torchaudio<=0.13",
-    "torchvision>=0.3.0,<=0.14",
 ]
-_pytorch_vision_deps = _pytorch_deps + ["torchvision>=0.3.0,<=0.14"]
+_pytorch_vision_deps = _pytorch_deps + ["torchvision>=0.3.0,<0.15"]
 _transformers_deps = _pytorch_deps + [
     f"{'nm-transformers' if is_release else 'nm-transformers-nightly'}"
     f"~={version_nm_deps}",