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

2025-12-21 22:53:31 -08:00
parent 1464b0dc8f
commit a9b1bb5947
1 changed files with 1 additions and 0 deletions
--- a/Dockerfile.wheels
+++ b/Dockerfile.wheels
@@ -81,6 +81,7 @@ ARG PRE_TRANSFORMERS=0
 RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \
    if [ "$PRE_TRANSFORMERS" = "1" ]; then \
        uv pip install -U transformers --pre; \
+        uv pip install numpy==2.2.6; \
    fi

 # Setup Env for Runtime