From 9f35dbdd2d3069443ff117b2f6a84483e384a4f5 Mon Sep 17 00:00:00 2001 From: Eugene Rakhmatulin Date: Sat, 20 Dec 2025 23:01:49 -0800 Subject: [PATCH] Reverted back to release flashinfer --- Dockerfile.wheels | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/Dockerfile.wheels b/Dockerfile.wheels index 4c6e964..da6ed1f 100644 --- a/Dockerfile.wheels +++ b/Dockerfile.wheels @@ -63,9 +63,9 @@ RUN patch -p1 -d /usr/local/lib/python3.12/dist-packages < ${VLLM_BASE_DIR}/fast # Install flashinfer helper packages RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \ - uv pip install --system --break-system-packages flashinfer-python -U --no-deps --pre --index-url https://flashinfer.ai/whl && \ - uv pip install --system --break-system-packages flashinfer-cubin --pre --index-url https://flashinfer.ai/whl && \ - uv pip install --system --break-system-packages flashinfer-jit-cache --pre --index-url https://flashinfer.ai/whl/cu130 + uv pip install --system --break-system-packages flashinfer-python -U --no-deps --index-url https://flashinfer.ai/whl && \ + uv pip install --system --break-system-packages flashinfer-cubin --index-url https://flashinfer.ai/whl && \ + uv pip install --system --break-system-packages flashinfer-jit-cache --index-url https://flashinfer.ai/whl/cu130 # Setup Env for Runtime ENV TORCH_CUDA_ARCH_LIST=12.1a