diff --git a/docker/Dockerfile b/docker/Dockerfile index b26b82eb598..5b6fcf026f4 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -682,6 +682,9 @@ RUN --mount=type=bind,from=build,src=/workspace/dist,target=/vllm-workspace/dist --extra-index-url ${PYTORCH_CUDA_INDEX_BASE_URL}/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.'); \ fi +RUN --mount=type=cache,target=/root/.cache/uv \ + uv pip install --system "transformers==5.4.0" + RUN --mount=type=cache,target=/root/.cache/uv \ . /etc/environment && \ uv pip list