Skip to content

Commit 1fbb019

Browse files
authored
Restrict setuptools version (#529)
* Restrict setuptools version Early access to vllm-project#17320 setuptools v80 broke too many things * Update Dockerfile.rocm_base
1 parent 8bd7ee1 commit 1fbb019

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

docker/Dockerfile.rocm_base

+1-1
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ RUN apt-get update -y \
4040
&& curl -sS https://bootstrap.pypa.io/get-pip.py | python${PYTHON_VERSION} \
4141
&& python3 --version && python3 -m pip --version
4242

43-
RUN pip install -U packaging 'cmake<4' ninja wheel setuptools pybind11 Cython
43+
RUN pip install -U packaging 'cmake<4' ninja wheel 'setuptools<80' pybind11 Cython
4444

4545
FROM base AS build_hipblaslt
4646
ARG HIPBLASLT_BRANCH

requirements/common.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ mistral_common[opencv] >= 1.5.4
3434
opencv-python-headless >= 4.11.0 # required for video IO
3535
pyyaml
3636
six>=1.16.0; python_version > '3.11' # transitive dependency of pandas that needs to be the latest version for python 3.12
37-
setuptools>=74.1.1; python_version > '3.11' # Setuptools is used by triton, we need to ensure a modern version is installed for 3.12+ so that it does not try to import distutils, which was removed in 3.12
37+
setuptools>=74.1.1,<80; python_version > '3.11' # Setuptools is used by triton, we need to ensure a modern version is installed for 3.12+ so that it does not try to import distutils, which was removed in 3.12
3838
einops # Required for Qwen2-VL.
3939
compressed-tensors == 0.9.3 # required for compressed-tensors
4040
depyf==0.18.0 # required for profiling and debugging with compilation config

0 commit comments

Comments
 (0)