Skip to content

Commit a7700c5

Browse files
dhuangnmdsikka
andauthored
Add upper bounds to dependencies for release (#1734)
SUMMARY: - added upper bounds to dependencies when build type is release - still allow latest dependency versions when build type is nightly The upper bounds are based on the current latest version of each dependency. May need to update again when we're ready to release 0.7.0. TEST PLAN: Run all tests. Kicked off nightly tests: https://github.com/neuralmagic/llm-compressor-testing/actions/runs/16976533641/job/48126802385 --------- Signed-off-by: Dan Huang <[email protected]> Co-authored-by: Dipika Sikka <[email protected]>
1 parent d1b00c7 commit a7700c5

File tree

1 file changed

+57
-13
lines changed

1 file changed

+57
-13
lines changed

setup.py

Lines changed: 57 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -110,23 +110,67 @@ def localversion_func(version: ScmVersion) -> str:
110110
"src", include=["llmcompressor", "llmcompressor.*"], exclude=["*.__pycache__.*"]
111111
),
112112
install_requires=[
113-
"loguru>=0.7.2",
114-
"pyyaml>=6.0.1",
113+
(
114+
"loguru>=0.7.2,<=0.7.3"
115+
if BUILD_TYPE == "release"
116+
else "loguru>=0.7.2"
117+
),
118+
(
119+
"pyyaml>=6.0.1,<=6.0.2"
120+
if BUILD_TYPE == "release"
121+
else "pyyaml>=6.0.1"
122+
),
115123
# librosa dependency numba is currently not compatible with numpy>=2.3
116124
# https://numba.readthedocs.io/en/stable/user/installing.html#version-support-information
117-
"numpy>=2.0.0,<2.3.0",
118-
"requests>=2.32.2",
119-
"tqdm>=4.66.3",
120-
"torch>=2.7.0",
121-
"transformers>=4.53.0",
122-
"datasets>=4.0.0",
123-
"accelerate>=1.6.0",
124-
"pynvml>=11.5.3",
125-
"pillow>=10.4.0",
126125
(
127-
"compressed-tensors==0.10.2"
126+
"numpy>=2.0.0,<=2.3.2"
127+
if BUILD_TYPE == "release"
128+
else "numpy>=2.0.0"
129+
),
130+
(
131+
"requests>=2.32.2,<=2.32.5"
132+
if BUILD_TYPE == "release"
133+
else "requests>=2.32.2"
134+
),
135+
(
136+
"tqdm>=4.66.3,<=4.67.1"
137+
if BUILD_TYPE == "release"
138+
else "tqdm>=4.66.3"
139+
),
140+
(
141+
"torch>=2.7.0,<=2.8.0"
142+
if BUILD_TYPE == "release"
143+
else "torch>=2.7.0"
144+
),
145+
(
146+
"transformers>=4.53.0,<=4.55.2"
147+
if BUILD_TYPE == "release"
148+
else "transformers>=4.53.0"
149+
),
150+
(
151+
"datasets>=4.0.0,<=4.0.0"
152+
if BUILD_TYPE == "release"
153+
else "datasets>=4.0.0"
154+
),
155+
(
156+
"accelerate>=1.6.0,<=1.10.0"
157+
if BUILD_TYPE == "release"
158+
else "accelerate>=1.6.0"
159+
),
160+
(
161+
"pynvml>=11.5.3,<=12.0.0"
162+
if BUILD_TYPE == "release"
163+
else "pynvml>=11.5.3"
164+
),
165+
(
166+
"pillow>=10.4.0,<=10.4.0"
167+
if BUILD_TYPE == "release"
168+
else "pillow>=10.4.0"
169+
),
170+
(
171+
"compressed-tensors==0.11.0"
128172
if BUILD_TYPE == "release"
129-
else "compressed-tensors>=0.10.3a2"
173+
else "compressed-tensors>=0.11.1a2"
130174
),
131175
],
132176
extras_require={

0 commit comments

Comments
 (0)