Skip to content

Commit d6548a8

Browse files
authored
Update upper bounds for some dependencies to be compatible with their new release (#1835)
SUMMARY: There are a few dependencies having new release lately and we want to be compatible. To be specific: numpy: https://pypi.org/project/numpy/2.3.3/ transformers: https://pypi.org/project/transformers/4.56.1/ datasets: https://pypi.org/project/datasets/4.1.0/ accelerate: https://pypi.org/project/accelerate/1.10.1/ pynvml: https://pypi.org/project/pynvml/13.0.1/ pillow: https://pypi.org/project/pillow/11.3.0/ Updated the upper bounds of these dependencies in the setup.py. TEST PLAN: Ran nightly tests with the latest version of these dependencies and tests passed: https://github.com/neuralmagic/llm-compressor-testing/actions/runs/17803108635/job/50607909450 --------- Signed-off-by: Dan Huang <[email protected]>
1 parent 495fcdf commit d6548a8

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

setup.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ def localversion_func(version: ScmVersion) -> str:
114114
("pyyaml>=6.0.1,<=6.0.2" if BUILD_TYPE == "release" else "pyyaml>=6.0.1"),
115115
# librosa dependency numba is currently not compatible with numpy>=2.3
116116
# https://numba.readthedocs.io/en/stable/user/installing.html#version-support-information
117-
("numpy>=2.0.0,<=2.3.2" if BUILD_TYPE == "release" else "numpy>=2.0.0"),
117+
("numpy>=2.0.0,<=2.3.3" if BUILD_TYPE == "release" else "numpy>=2.0.0"),
118118
(
119119
"requests>=2.32.2,<=2.32.5"
120120
if BUILD_TYPE == "release"
@@ -123,18 +123,18 @@ def localversion_func(version: ScmVersion) -> str:
123123
("tqdm>=4.66.3,<=4.67.1" if BUILD_TYPE == "release" else "tqdm>=4.66.3"),
124124
("torch>=2.7.0,<=2.8.0" if BUILD_TYPE == "release" else "torch>=2.7.0"),
125125
(
126-
"transformers>=4.53.0,<=4.55.2"
126+
"transformers>=4.53.0,<=4.56.1"
127127
if BUILD_TYPE == "release"
128128
else "transformers>=4.53.0"
129129
),
130-
("datasets>=4.0.0,<=4.0.0" if BUILD_TYPE == "release" else "datasets>=4.0.0"),
130+
("datasets>=4.0.0,<=4.1.0" if BUILD_TYPE == "release" else "datasets>=4.0.0"),
131131
(
132-
"accelerate>=1.6.0,<=1.10.0"
132+
"accelerate>=1.6.0,<=1.10.1"
133133
if BUILD_TYPE == "release"
134134
else "accelerate>=1.6.0"
135135
),
136-
("pynvml>=11.5.3,<=12.0.0" if BUILD_TYPE == "release" else "pynvml>=11.5.3"),
137-
("pillow>=10.4.0,<=10.4.0" if BUILD_TYPE == "release" else "pillow>=10.4.0"),
136+
("pynvml>=11.5.3,<=13.0.1" if BUILD_TYPE == "release" else "pynvml>=11.5.3"),
137+
("pillow>=10.4.0,<=11.3.0" if BUILD_TYPE == "release" else "pillow>=10.4.0"),
138138
(
139139
"compressed-tensors==0.11.0"
140140
if BUILD_TYPE == "release"

0 commit comments

Comments
 (0)