Revert "update deps"

This reverts commit bed8bb81.

Revert "update deps"
This reverts commit bed8bb81.
02845513 · zhuwenwen · bed8bb81 · 02845513 · 02845513
Commit 02845513 authored Jan 09, 2026 by zhuwenwen
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

pyproject.toml pyproject.toml +1 -1

requirements/rocm.txt requirements/rocm.txt +2 -2

No files found.
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -6,7 +6,7 @@ requires = [
    "packaging>=24.2",
    "setuptools>=77.0.3,<80.0.0",
    "setuptools-scm>=8.0",
-    "torch==2.5.1",
+    "torch>=2.5.1,<=2.7.1",
    "wheel",
    "jinja2",
 ]

--- a/requirements/rocm.txt
+++ b/requirements/rocm.txt
@@ -28,8 +28,8 @@ quart
 fastrlock==0.8.3
 cupy==12.3.0

-torch==2.5.1
-torchvision==0.20.1
+torch>=2.5.1,<=2.7.1
+torchvision>=0.20.1,<=0.22.0
 triton == 3.1
 flash_attn == 2.6.1
 flash_mla == 1.0.0