From 0803bf18fd07cd7429cb6ea54fe0dccbb469022d Mon Sep 17 00:00:00 2001 From: khluu Date: Thu, 2 Apr 2026 11:47:14 -0700 Subject: [PATCH] p Signed-off-by: khluu --- .buildkite/scripts/generate-and-upload-nightly-index.sh | 6 ++---- docker/Dockerfile.rocm | 4 +++- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/.buildkite/scripts/generate-and-upload-nightly-index.sh b/.buildkite/scripts/generate-and-upload-nightly-index.sh index fa6eb979af5..7289003ee5f 100755 --- a/.buildkite/scripts/generate-and-upload-nightly-index.sh +++ b/.buildkite/scripts/generate-and-upload-nightly-index.sh @@ -53,10 +53,8 @@ echo "Uploading indices to $S3_COMMIT_PREFIX" aws s3 cp --recursive "$INDICES_OUTPUT_DIR/" "$S3_COMMIT_PREFIX" # copy to /nightly/ only if it is on the main branch and not a PR -if [[ "$BUILDKITE_BRANCH" == "main" && "$BUILDKITE_PULL_REQUEST" == "false" ]]; then - echo "Uploading indices to overwrite /nightly/" - aws s3 cp --recursive "$INDICES_OUTPUT_DIR/" "s3://$BUCKET/nightly/" -fi +echo "Uploading indices to overwrite /nightly/" +aws s3 cp --recursive "$INDICES_OUTPUT_DIR/" "s3://$BUCKET/nightly/" # detect version from any wheel in the commit directory # download the first wheel we find to extract version metadata diff --git a/docker/Dockerfile.rocm b/docker/Dockerfile.rocm index 6db6d8b8359..08747de9769 100644 --- a/docker/Dockerfile.rocm +++ b/docker/Dockerfile.rocm @@ -330,7 +330,8 @@ RUN --mount=type=bind,from=export_vllm,src=/,target=/install \ && uv pip install --system -r requirements/rocm.txt \ && uv pip install --system -r requirements/rocm-test.txt \ && pip uninstall -y vllm \ - && uv pip install --system *.whl + && uv pip install --system *.whl \ + && uv pip install --system "transformers==5.5.0" # Verify that PyTorch is the ROCm build, not CUDA RUN python3 -c "import torch; assert torch.version.hip is not None, \ @@ -428,6 +429,7 @@ RUN --mount=type=bind,from=export_vllm,src=/,target=/install \ && uv pip install --system -r requirements/rocm.txt \ && pip uninstall -y vllm \ && uv pip install --system *.whl + && uv pip install --system "transformers==5.5.0" ARG COMMON_WORKDIR ARG BASE_IMAGE