update deps

bed8bb81 · zhuwenwen · e00ed52f · bed8bb81 · bed8bb81
Commit bed8bb81 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,<=2.7.1",
+    "torch==2.5.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,<=2.7.1
-torchvision>=0.20.1,<=0.22.0
+torch==2.5.1
+torchvision==0.20.1
 triton == 3.1
 flash_attn == 2.6.1
 flash_mla == 1.0.0