Support torch 2.3 and remove version ceiling restriction

neuralmagic · May 2, 2024 · fd9d091 · fd9d091
1 parent ee0be47
commit fd9d091
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -125,9 +125,9 @@
 _onnxruntime_deps = [
     "onnxruntime>=1.7.0",
 ]
-_torch_deps = ["torch>=1.7.0,<2.2"]
+_torch_deps = ["torch>=1.7.0"]
 _computer_vision_deps = [
-    "torchvision>=0.3.0,<0.17",
+    "torchvision>=0.3.0",
     "opencv-python<=4.6.0.66",
 ]
 _openpifpaf_integration_deps = [