Skip to content

GitLab

Commit b4bb5ea6 authored by zhuwenwen's avatar zhuwenwen
Browse files

update torchvision

parent 4b4a966d
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
pyproject.toml
View file @ b4bb5ea6
...@@ -6,7 +6,7 @@ requires = [ ...@@ -6,7 +6,7 @@ requires = [
"packaging>=24.2", "packaging>=24.2",
"setuptools>=77.0.3,<80.0.0", "setuptools>=77.0.3,<80.0.0",
"setuptools-scm>=8.0", "setuptools-scm>=8.0",
"torch >= 2.5.1", "torch>=2.5.1,<=2.7.1",
"wheel", "wheel",
"jinja2", "jinja2",
] ]
......
requirements/rocm.txt
View file @ b4bb5ea6
...@@ -28,7 +28,7 @@ quart ...@@ -28,7 +28,7 @@ quart
fastrlock==0.8.3 fastrlock==0.8.3
cupy==12.3.0 cupy==12.3.0
torch >= 2.5.1 torch>=2.5.1,<=2.7.1
triton == 3.1 triton == 3.1
flash_attn == 2.6.1 flash_attn == 2.6.1
flash_mla == 1.0.0 flash_mla == 1.0.0
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment