From edd435f0f9bb042dd0f56990b8441f5b87bc9963 Mon Sep 17 00:00:00 2001
From: Michael Goin <michael@neuralmagic.com>
Date: Mon, 6 May 2024 11:09:38 -0400
Subject: [PATCH] Support torch 2.3 and remove version ceiling restriction
 (#1646)

---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index c58dd49bfd..d030473bfd 100644
--- 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 = [