Skip to content

Commit efe58e0

Browse files
committed
install blobfile for Kimi model support
1 parent 73b8071 commit efe58e0

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

Dockerfile.rocm.ubi

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -189,7 +189,8 @@ RUN --mount=type=bind,from=build_amdsmi,src=/install,target=/install/amdsmi/ \
189189
--extra-index-url "https://download.pytorch.org/whl/nightly/rocm${version}" \
190190
/install/amdsmi/*.whl\
191191
/install/flashattention/*.whl\
192-
"$(echo /install/vllm/*.whl)[audio,video,tensorizer]"
192+
"$(echo /install/vllm/*.whl)[audio,video,tensorizer]" \
193+
&& uv pip install blobfile
193194

194195
ENV HF_HUB_OFFLINE=1 \
195196
HOME=/home/vllm \

Dockerfile.ubi

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,7 @@ RUN --mount=type=bind,from=build,src=/workspace/dist,target=/workspace/dist \
145145
--extra-index-url="https://download.pytorch.org/whl/cu128" --index-strategy='unsafe-best-match' \
146146
"$(echo dist/*.whl)[audio,video,tensorizer]" --verbose \
147147
"https://storage.googleapis.com/nm-public-pypi/dist/flashinfer_python-0.2.8-cp39-abi3-linux_x86_64.whl" \
148-
&& uv pip install -U nvidia-nccl-cu12==2.26.5
148+
&& uv pip install -U nvidia-nccl-cu12==2.26.5 blobfile
149149

150150
ENV HF_HUB_OFFLINE=1 \
151151
HOME=/home/vllm \

0 commit comments

Comments
 (0)