From a9b1bb5947feac4532ee992ac2a081a302503a7d Mon Sep 17 00:00:00 2001 From: Eugene Rakhmatulin Date: Sun, 21 Dec 2025 22:53:31 -0800 Subject: [PATCH] fixed a bug with numpy version in wheels build when transformers 5 is used. --- Dockerfile.wheels | 1 + 1 file changed, 1 insertion(+) diff --git a/Dockerfile.wheels b/Dockerfile.wheels index 8fa30da..a224bb1 100644 --- 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