Skip to content

Commit bf9fd89

Browse files
committed
vLLM: take the CPU wheel from the release's assets
The latest release is tagged v0.33.0rc3, but its wheels are named 0.33.0, so the URL built from the tag answered 404 and the CPU install stopped (2 runs). The wheel now comes from the release's asset list.
1 parent d63417a commit bf9fd89

2 files changed

Lines changed: 16 additions & 6 deletions

File tree

‎ct/vllm.sh‎

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -50,9 +50,14 @@ function update_script() {
5050
--extra-index-url https://wheels.vllm.ai/rocm/
5151
;;
5252
cpu)
53-
$STD uv pip install --python /opt/vllm/bin/python \
54-
"https://github.com/vllm-project/vllm/releases/download/v${VLLM_VERSION}/vllm-${VLLM_VERSION}+cpu-cp38-abi3-manylinux_2_39_x86_64.whl" \
55-
--torch-backend cpu
53+
github_api_call "https://api.github.com/repos/vllm-project/vllm/releases/tags/v${VLLM_VERSION}" /tmp/vllm-release.json
54+
VLLM_WHEEL="$(jq -r '.assets[] | select(.name | endswith("+cpu-cp38-abi3-manylinux_2_39_x86_64.whl")) | .browser_download_url' /tmp/vllm-release.json | head -n1)"
55+
rm -f /tmp/vllm-release.json
56+
[[ -n "$VLLM_WHEEL" ]] || {
57+
msg_error "vLLM ${VLLM_VERSION} has no CPU wheel for x86_64"
58+
exit 1
59+
}
60+
$STD uv pip install --python /opt/vllm/bin/python "$VLLM_WHEEL" --torch-backend cpu
5661
;;
5762
*)
5863
$STD uv pip install --python /opt/vllm/bin/python --upgrade vllm

‎install/vllm-install.sh‎

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -56,9 +56,14 @@ cpu)
5656
msg_error "vLLM publishes no prebuilt CPU wheel for $(uname -m)"
5757
exit 1
5858
fi
59-
$STD uv pip install --python /opt/vllm/bin/python \
60-
"https://github.com/vllm-project/vllm/releases/download/v${VLLM_VERSION}/vllm-${VLLM_VERSION}+cpu-cp38-abi3-manylinux_2_39_x86_64.whl" \
61-
--torch-backend cpu
59+
github_api_call "https://api.github.com/repos/vllm-project/vllm/releases/tags/v${VLLM_VERSION}" /tmp/vllm-release.json
60+
VLLM_WHEEL="$(jq -r '.assets[] | select(.name | endswith("+cpu-cp38-abi3-manylinux_2_39_x86_64.whl")) | .browser_download_url' /tmp/vllm-release.json | head -n1)"
61+
rm -f /tmp/vllm-release.json
62+
[[ -n "$VLLM_WHEEL" ]] || {
63+
msg_error "vLLM ${VLLM_VERSION} has no CPU wheel for x86_64"
64+
exit 1
65+
}
66+
$STD uv pip install --python /opt/vllm/bin/python "$VLLM_WHEEL" --torch-backend cpu
6267
;;
6368
*)
6469
msg_error "Unknown var_backend '${var_backend}' (expected auto, cuda, rocm or cpu)"

0 commit comments

Comments
 (0)