Update pyproject.toml

pytorch · Jan 7, 2025 · 48dce20 · 48dce20
1 parent 4ee017d
commit 48dce20
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -9,7 +9,7 @@ requires = [
     "typing-extensions>=4.7.0",
     "future>=0.18.3",
     "tensorrt-cu12>=10.6.0,<10.8.0",
-    "torch>=2.6.0,<2.7.0",
+    "torch>=2.6.0.dev,<2.7.0",
     "pybind11==2.6.2",
     "numpy",
 ]
@@ -54,7 +54,7 @@ keywords = [
     "inference",
 ]
 dependencies = [
-    "torch>=2.6.0,<2.7.0",
+    "torch>=2.6.0.dev,<2.7.0",
     "tensorrt-cu12>=10.6.0,<10.8.0",
     "tensorrt-cu12-bindings>=10.6.0,<10.8.0",
     "tensorrt-cu12-libs>=10.6.0,<10.8.0",