Skip to content

Commit 05a13f3

Browse files
authored
Support torch 2.10 (#2356)
SUMMARY: - Requires: vllm-project/compressed-tensors#583 - Transformers tests currently already failing on main
1 parent c37fcfa commit 05a13f3

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

setup.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ def localversion_func(version: ScmVersion) -> str:
121121
else "requests>=2.32.2"
122122
),
123123
("tqdm>=4.66.3,<=4.67.1" if BUILD_TYPE == "release" else "tqdm>=4.66.3"),
124-
("torch>=2.9.0,<=2.9.1" if BUILD_TYPE == "release" else "torch>=2.9.0,<=2.9.1"),
124+
("torch>=2.9.0,<=2.10.0" if BUILD_TYPE == "release" else "torch>=2.9.0"),
125125
(
126126
"transformers>=4.56.1,<=4.57.6"
127127
if BUILD_TYPE == "release"
@@ -163,10 +163,10 @@ def localversion_func(version: ScmVersion) -> str:
163163
"cmarkgfm>=2024.1.14",
164164
"trl>=0.10.1",
165165
"pandas<2.3.0",
166-
"torchvision<=0.24.1",
166+
"torchvision",
167167
"librosa==0.11.0",
168168
"soundfile",
169-
"torchcodec<=0.9.1",
169+
"torchcodec",
170170
# linting, formatting, and type checking
171171
"mypy~=1.10.0",
172172
"ruff~=0.4.8",

0 commit comments

Comments
 (0)