Dockerfile.ubi: bump flashinfer to 0.1.2

opendatahub-io · Aug 19, 2024 · 9460cfb · 9460cfb
1 parent ac48a82
commit 9460cfb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile.ubi b/Dockerfile.ubi
@@ -178,7 +178,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.0.9/flashinfer-0.0.9+cu121torch2.3-cp311-cp311-linux_x86_64.whl
+    uv pip install https://github.com/flashinfer-ai/flashinfer/releases/download/v0.1.2/flashinfer-0.1.2+cu121torch2.4-cp311-cp311-linux_x86_64.whl
 
 ENV HF_HUB_OFFLINE=1 \
     PORT=8000 \