From 9460cfb8b869199c7631eb438ec78255e8b5a6d9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Daniele=20Trifir=C3=B2?= Date: Mon, 5 Aug 2024 18:17:52 +0200 Subject: [PATCH] Dockerfile.ubi: bump flashinfer to 0.1.2 --- Dockerfile.ubi | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile.ubi b/Dockerfile.ubi index 914cae070274b..53a6c90ecfbf9 100644 --- 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 \