Dockerfile.ubi: bump flashinfer to v0.2.0.post2

https://github.com/flashinfer-ai/flashinfer/releases/tag/v0.2.0.post2
opendatahub-io · Feb 7, 2025 · ae28d3e · ae28d3e
1 parent 8f6761b
commit ae28d3e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile.ubi b/Dockerfile.ubi
@@ -174,7 +174,7 @@ RUN --mount=type=bind,from=libsodium-builder,src=/usr/src/libsodium,target=/usr/
 RUN --mount=type=cache,target=/root/.cache/pip \
     --mount=type=cache,target=/root/.cache/uv \
     uv pip install \
-        "https://github.com/flashinfer-ai/flashinfer/releases/download/v0.1.6/flashinfer-0.1.6+cu124torch2.4-cp312-cp312-linux_x86_64.whl"
+        "https://github.com/flashinfer-ai/flashinfer/releases/download/v0.2.0.post2/flashinfer_python-0.2.0.post2+cu124torch2.5-cp312-cp312-linux_x86_64.whl"
 
 ENV HF_HUB_OFFLINE=1 \
     HOME=/home/vllm \