karylab_agents/vllm
1
0
Fork 0
forked from Karylab-cklius/vllm
Code Pull Requests 2 Actions 1 Packages Activity

p

Browse Source 
Signed-off-by: khluu <khluu000@gmail.com>
This commit is contained in:
khluu
2026-04-02 11:47:14 -07:00
parent 08ed2b9688
commit 0803bf18fd
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.buildkite/scripts/generate-and-upload-nightly-index.sh
+2 -4
View File
@@ -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
+3 -1
View File
@@ -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