software-engineering/spark-vllm-docker
2
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

fixed a bug with numpy version in wheels build when transformers 5 is used.

Browse Source
This commit is contained in:
Eugene Rakhmatulin
2025-12-21 22:53:31 -08:00
parent 1464b0dc8f
commit a9b1bb5947
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
Dockerfile.wheels
View File

@@ -81,6 +81,7 @@ ARG PRE_TRANSFORMERS=0
RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \ RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \
if [ "$PRE_TRANSFORMERS" = "1" ]; then \ if [ "$PRE_TRANSFORMERS" = "1" ]; then \
uv pip install -U transformers --pre; \ uv pip install -U transformers --pre; \
uv pip install numpy==2.2.6; \
fi fi
# Setup Env for Runtime # Setup Env for Runtime