p

Signed-off-by: khluu <khluu000@gmail.com>

.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

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, \
@@ -427,7 +428,8 @@ RUN --mount=type=bind,from=export_vllm,src=/,target=/install \
     cd /install \
     && uv pip install --system -r requirements/rocm.txt \
     && pip uninstall -y vllm \
-    && uv pip install --system *.whl
+    && uv pip install --system *.whl \
+    && uv pip install --system "transformers==5.5.0"
 
 ARG COMMON_WORKDIR
 ARG BASE_IMAGE