Skip to content

Commit 12aa563

Browse files
dhuangnmdsikka
andauthored
[WIP] Update dependency bounds for new release (vllm-project#2407)
SUMMARY: Update llmcompressor dependency bounds except for compressed-tensors, which will be updated after the compressed-tensors 0.14.0 is released. TEST PLAN: All tests --------- Signed-off-by: Dan Huang <dahuang@redhat.com> Co-authored-by: Dipika Sikka <dipikasikka1@gmail.com>
1 parent 81ec39c commit 12aa563

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

setup.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -112,25 +112,25 @@ def localversion_func(version: ScmVersion) -> str:
112112
install_requires=[
113113
("loguru>=0.7.2,<=0.7.3" if BUILD_TYPE == "release" else "loguru>=0.7.2"),
114114
("pyyaml>=6.0.1,<=6.0.3" if BUILD_TYPE == "release" else "pyyaml>=6.0.1"),
115-
# librosa dependency numba is currently not compatible with numpy>=2.3
116-
# https://numba.readthedocs.io/en/stable/user/installing.html#version-support-information
117-
("numpy>=2.0.0,<=2.3.5" if BUILD_TYPE == "release" else "numpy>=2.0.0"),
115+
# librosa 0.11.0 supports numpy 2.x
116+
# https://librosa.org/doc/0.11.0/changelog.html
117+
("numpy>=2.0.0,<=2.4.2" if BUILD_TYPE == "release" else "numpy>=2.0.0"),
118118
(
119119
"requests>=2.32.2,<=2.32.5"
120120
if BUILD_TYPE == "release"
121121
else "requests>=2.32.2"
122122
),
123-
("tqdm>=4.66.3,<=4.67.1" if BUILD_TYPE == "release" else "tqdm>=4.66.3"),
123+
("tqdm>=4.66.3,<=4.67.3" if BUILD_TYPE == "release" else "tqdm>=4.66.3"),
124124
("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"
128128
else "transformers>=4.56.1,<=4.57.6"
129129
),
130-
("datasets>=4.0.0,<=4.4.1" if BUILD_TYPE == "release" else "datasets>=4.0.0"),
130+
("datasets>=4.0.0,<=4.6.0" if BUILD_TYPE == "release" else "datasets>=4.0.0"),
131131
(
132132
# auto-round 0.9.1 cannot work with accelerate <1.10.0
133-
"auto-round>=0.9.6,<=0.9.6"
133+
"auto-round>=0.9.6,<=0.10.2"
134134
if BUILD_TYPE == "release"
135135
else "auto-round>=0.9.6"
136136
),
@@ -140,11 +140,11 @@ def localversion_func(version: ScmVersion) -> str:
140140
else "accelerate>=1.6.0"
141141
),
142142
(
143-
"nvidia-ml-py>=12.560.30,<=13.590.44"
143+
"nvidia-ml-py>=12.560.30,<=13.590.48"
144144
if BUILD_TYPE == "release"
145145
else "nvidia-ml-py>=12.560.30"
146146
),
147-
("pillow>=10.4.0,<=12.0.0" if BUILD_TYPE == "release" else "pillow>=10.4.0"),
147+
("pillow>=10.4.0,<=12.1.1" if BUILD_TYPE == "release" else "pillow>=10.4.0"),
148148
(
149149
"compressed-tensors==0.13.0"
150150
if BUILD_TYPE == "release"

0 commit comments

Comments
 (0)