From 7bd3f40dda2a2918a7dc83d9b36ebe1d9ed3d625 Mon Sep 17 00:00:00 2001 From: "Li, Jiang" Date: Sat, 11 Apr 2026 15:04:25 +0800 Subject: [PATCH] [CI/Build] Fix sentence-transformers version in CPU test (#39557) Signed-off-by: jiang1.li (cherry picked from commit eb92ba740a63c2ae590ccfb16f07b490d261d30a) Signed-off-by: khluu --- docker/Dockerfile.cpu | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/docker/Dockerfile.cpu b/docker/Dockerfile.cpu index 6582b02f0b9..4df522ac371 100644 --- a/docker/Dockerfile.cpu +++ b/docker/Dockerfile.cpu @@ -140,10 +140,12 @@ RUN \ esac; \ }; \ remove_packages_not_supported_on_aarch64 && \ - sed -i 's/^torch==.*/torch==2.10.0/g' requirements/cpu-test.in && \ - sed -i 's/torchaudio.*/torchaudio/g' requirements/cpu-test.in && \ - sed -i 's/torchvision.*/torchvision/g' requirements/cpu-test.in && \ - uv pip compile requirements/cpu-test.in -o requirements/cpu-test.txt --index-strategy unsafe-best-match --torch-backend cpu + sed -i 's/^torch==.*/torch==2.11.0/g' requirements/test/cpu.in && \ + sed -i 's/torchaudio.*/torchaudio/g' requirements/test/cpu.in && \ + sed -i 's/torchvision.*/torchvision/g' requirements/test/cpu.in && \ + # Related issue: https://github.com/vllm-project/vllm/pull/38800#issuecomment-4228314305 + sed -i 's/^sentence-transformers.*/sentence-transformers==5.3.0/g' requirements/test/cpu.in && \ + uv pip compile requirements/test/cpu.in -o requirements/test/cpu.txt --index-strategy unsafe-best-match --torch-backend cpu RUN --mount=type=cache,target=/root/.cache/uv \ uv pip install -r requirements/cpu-test.txt