Skip to content

Commit 29835bc

Browse files
committed
update flashinfer wheel 0.2.5 location to latest
1 parent 7311fbd commit 29835bc

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

Dockerfile.ubi

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,7 @@ RUN --mount=type=bind,from=build,src=/workspace/dist,target=/workspace/dist \
138138
uv pip install \
139139
--extra-index-url="https://download.pytorch.org/whl/cu128" --index-strategy='unsafe-best-match' \
140140
"$(echo dist/*.whl)[audio,video,tensorizer]" --verbose \
141-
"https://storage.googleapis.com/neuralmagic-public-pypi/dist/flashinfer_python-0.2.5-cp38-abi3-linux_x86_64.whl"
141+
"https://storage.googleapis.com/nm-public-pypi/dist/flashinfer_python-0.2.5-cp38-abi3-linux_x86_64.whl"
142142

143143
ENV HF_HUB_OFFLINE=1 \
144144
HOME=/home/vllm \
@@ -196,4 +196,4 @@ ENV GRPC_PORT=8033 \
196196
DISABLE_LOGPROBS_DURING_SPEC_DECODING=false
197197

198198
USER 2000
199-
ENTRYPOINT ["python3", "-m", "vllm_tgis_adapter", "--uvicorn-log-level=warning"]
199+
ENTRYPOINT ["python3", "-m", "vllm_tgis_adapter", "--uvicorn-log-level=warning"]

0 commit comments

Comments
 (0)