Update docker/Dockerfile

Signed-off-by: Pavani Majety <pavanimajety@gmail.com>
This commit is contained in:
Pavani Majety 2025-12-22 10:55:19 -08:00 committed by GitHub
parent 20d87ee2f0
commit 4f1aa77ff9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -466,7 +466,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
# Install FlashInfer pre-compiled kernel cache and binaries
# This is ~1.1GB and only changes when FlashInfer version bumps
# https://docs.flashinfer.ai/installation.html
ARG FLASHINFER_VERSION=0.6.0rc1
ARG FLASHINFER_VERSION=0.6.0rc2
RUN --mount=type=cache,target=/root/.cache/uv \
uv pip install --system flashinfer-cubin==${FLASHINFER_VERSION} \
--extra-index-url https://flashinfer.ai/whl \