From fe7b04af58b9d2296ce976a9441982387064b96b Mon Sep 17 00:00:00 2001 From: Dan Huang Date: Tue, 2 Dec 2025 10:33:39 -0500 Subject: [PATCH 1/7] update upper bounds for 0.9 release Signed-off-by: Dan Huang --- setup.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/setup.py b/setup.py index 79325409c..f4f278035 100644 --- a/setup.py +++ b/setup.py @@ -114,22 +114,22 @@ def localversion_func(version: ScmVersion) -> str: ("pyyaml>=6.0.1,<=6.0.3" if BUILD_TYPE == "release" else "pyyaml>=6.0.1"), # librosa dependency numba is currently not compatible with numpy>=2.3 # https://numba.readthedocs.io/en/stable/user/installing.html#version-support-information - ("numpy>=2.0.0,<=2.3.3" if BUILD_TYPE == "release" else "numpy>=2.0.0"), + ("numpy>=2.0.0,<=2.3.5" if BUILD_TYPE == "release" else "numpy>=2.0.0"), ( "requests>=2.32.2,<=2.32.5" if BUILD_TYPE == "release" else "requests>=2.32.2" ), ("tqdm>=4.66.3,<=4.67.1" if BUILD_TYPE == "release" else "tqdm>=4.66.3"), - ("torch>=2.7.0,<=2.8.0" if BUILD_TYPE == "release" else "torch>=2.7.0"), + ("torch>=2.7.0,<=2.9.1" if BUILD_TYPE == "release" else "torch>=2.7.0"), ( - "transformers>=4.54.0,<=4.56.2" + "transformers>=4.54.0,<=4.57.3" if BUILD_TYPE == "release" else "transformers>=4.54.0" ), - ("datasets>=4.0.0,<=4.1.1" if BUILD_TYPE == "release" else "datasets>=4.0.0"), + ("datasets>=4.0.0,<=4.4.1" if BUILD_TYPE == "release" else "datasets>=4.0.0"), ( - "accelerate>=1.6.0,<=1.10.1" + "accelerate>=1.6.0,<=1.12.0" if BUILD_TYPE == "release" else "accelerate>=1.6.0" ), @@ -138,13 +138,13 @@ def localversion_func(version: ScmVersion) -> str: if BUILD_TYPE == "release" else "nvidia-ml-py>=12.560.30" ), - ("pillow>=10.4.0,<=11.3.0" if BUILD_TYPE == "release" else "pillow>=10.4.0"), + ("pillow>=10.4.0,<=12.0.0" if BUILD_TYPE == "release" else "pillow>=10.4.0"), ( "compressed-tensors==0.12.2" if BUILD_TYPE == "release" else "compressed-tensors>=0.12.3a2" ), - ("auto-round==0.9.1"), + ("auto-round==0.9.1" if BUILD_TYPE == "release" else "auto-round==0.9.1"), ], extras_require={ "dev": [ From 09baec83c2c584c51956fb379a4cddca09a193be Mon Sep 17 00:00:00 2001 From: Dan Huang Date: Tue, 2 Dec 2025 12:06:56 -0500 Subject: [PATCH 2/7] update accelerate based on auto-round requirement Signed-off-by: Dan Huang --- setup.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index f4f278035..f5ef3288c 100644 --- a/setup.py +++ b/setup.py @@ -128,8 +128,9 @@ def localversion_func(version: ScmVersion) -> str: else "transformers>=4.54.0" ), ("datasets>=4.0.0,<=4.4.1" if BUILD_TYPE == "release" else "datasets>=4.0.0"), + # auto-round 0.9.1 requires accelerate >= 1.10.0 ( - "accelerate>=1.6.0,<=1.12.0" + "accelerate>=1.10.0,<=1.12.0" if BUILD_TYPE == "release" else "accelerate>=1.6.0" ), From 19fb2b2d06fbf36ea9d7bc6ee8531048932e8d0a Mon Sep 17 00:00:00 2001 From: Dan Huang Date: Tue, 2 Dec 2025 15:31:20 -0500 Subject: [PATCH 3/7] update transformers and auto-round Signed-off-by: Dan Huang --- setup.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index f5ef3288c..acba9ae39 100644 --- a/setup.py +++ b/setup.py @@ -123,7 +123,7 @@ def localversion_func(version: ScmVersion) -> str: ("tqdm>=4.66.3,<=4.67.1" if BUILD_TYPE == "release" else "tqdm>=4.66.3"), ("torch>=2.7.0,<=2.9.1" if BUILD_TYPE == "release" else "torch>=2.7.0"), ( - "transformers>=4.54.0,<=4.57.3" + "transformers>=4.54.0,<=5.0.0" if BUILD_TYPE == "release" else "transformers>=4.54.0" ), @@ -145,9 +145,11 @@ def localversion_func(version: ScmVersion) -> str: if BUILD_TYPE == "release" else "compressed-tensors>=0.12.3a2" ), - ("auto-round==0.9.1" if BUILD_TYPE == "release" else "auto-round==0.9.1"), ], extras_require={ + "autoround": [ + "auto-round==0.9.1", + ], "dev": [ # testing framework "pytest>=6.0.0", From c79b83c089983e75174ca1add211637eff710007 Mon Sep 17 00:00:00 2001 From: Dan Huang Date: Tue, 2 Dec 2025 15:36:42 -0500 Subject: [PATCH 4/7] fix format Signed-off-by: Dan Huang --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index acba9ae39..5181bec93 100644 --- a/setup.py +++ b/setup.py @@ -181,7 +181,7 @@ def localversion_func(version: ScmVersion) -> str: "mkdocstrings-python", "mkdocs-gen-files", "mkdocs-nav-weight", - ] + ], }, entry_points={ "console_scripts": [ From be3a99f55bc3eac755c6cda743c18b86875b392b Mon Sep 17 00:00:00 2001 From: Dan Huang Date: Tue, 2 Dec 2025 16:32:15 -0500 Subject: [PATCH 5/7] update transformers Signed-off-by: Dan Huang --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 5181bec93..bb9893f03 100644 --- a/setup.py +++ b/setup.py @@ -123,7 +123,7 @@ def localversion_func(version: ScmVersion) -> str: ("tqdm>=4.66.3,<=4.67.1" if BUILD_TYPE == "release" else "tqdm>=4.66.3"), ("torch>=2.7.0,<=2.9.1" if BUILD_TYPE == "release" else "torch>=2.7.0"), ( - "transformers>=4.54.0,<=5.0.0" + "transformers>=4.54.0,<=4.57.3" if BUILD_TYPE == "release" else "transformers>=4.54.0" ), From 71b547426ecd288859155df207bbd35c340bd176 Mon Sep 17 00:00:00 2001 From: Dan Huang Date: Tue, 2 Dec 2025 16:35:17 -0500 Subject: [PATCH 6/7] update transformers test workflow Signed-off-by: Dan Huang --- .github/workflows/test-check-transformers.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test-check-transformers.yaml b/.github/workflows/test-check-transformers.yaml index 4753e9145..dcefebccb 100644 --- a/.github/workflows/test-check-transformers.yaml +++ b/.github/workflows/test-check-transformers.yaml @@ -70,7 +70,7 @@ jobs: - name: Install uv uses: astral-sh/setup-uv@v6 - name: "⚙️ Install dependencies" - run: uv pip install .[dev] + run: uv pip install .[dev,autoround] - uses: actions/checkout@v4 with: repository: "vllm-project/compressed-tensors" From f43d46f3badbff717900fca97077400182d924a5 Mon Sep 17 00:00:00 2001 From: Dan Huang Date: Wed, 3 Dec 2025 11:15:30 -0500 Subject: [PATCH 7/7] revert autoround changes Signed-off-by: Dan Huang --- .github/workflows/test-check-transformers.yaml | 2 +- setup.py | 4 +--- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/.github/workflows/test-check-transformers.yaml b/.github/workflows/test-check-transformers.yaml index dcefebccb..4753e9145 100644 --- a/.github/workflows/test-check-transformers.yaml +++ b/.github/workflows/test-check-transformers.yaml @@ -70,7 +70,7 @@ jobs: - name: Install uv uses: astral-sh/setup-uv@v6 - name: "⚙️ Install dependencies" - run: uv pip install .[dev,autoround] + run: uv pip install .[dev] - uses: actions/checkout@v4 with: repository: "vllm-project/compressed-tensors" diff --git a/setup.py b/setup.py index bb9893f03..7bdb31ffb 100644 --- a/setup.py +++ b/setup.py @@ -128,6 +128,7 @@ def localversion_func(version: ScmVersion) -> str: else "transformers>=4.54.0" ), ("datasets>=4.0.0,<=4.4.1" if BUILD_TYPE == "release" else "datasets>=4.0.0"), + ("auto-round==0.9.1" if BUILD_TYPE == "release" else "auto-round==0.9.1"), # auto-round 0.9.1 requires accelerate >= 1.10.0 ( "accelerate>=1.10.0,<=1.12.0" @@ -147,9 +148,6 @@ def localversion_func(version: ScmVersion) -> str: ), ], extras_require={ - "autoround": [ - "auto-round==0.9.1", - ], "dev": [ # testing framework "pytest>=6.0.0",