File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 371371# Install FlashInfer pre-compiled kernel cache and binaries
372372# https://docs.flashinfer.ai/installation.html
373373RUN --mount=type=cache,target=/root/.cache/uv \
374- uv pip install --system flashinfer-cubin==0.4.1 \
375- && uv pip install --system flashinfer-jit-cache==0.4.1 \
374+ uv pip install --system flashinfer-cubin==0.5.0rc1 \
375+ && uv pip install --system flashinfer-jit-cache==0.5.0rc1 \
376376 --extra-index-url https://flashinfer.ai/whl/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.' ) \
377377 && flashinfer show-config
378378
Original file line number Diff line number Diff line change @@ -12,4 +12,4 @@ torchvision==0.24.0 # Required for phi3v processor. See https://github.com/pytor
1212# https://github.com/facebookresearch/xformers/releases/tag/v0.0.32.post1
1313# xformers==0.0.32.post1; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch >= 2.8
1414# FlashInfer should be updated together with the Dockerfile
15- flashinfer-python==0.4.1
15+ flashinfer-python==0.5.0rc1
You can’t perform that action at this time.
0 commit comments