From 160b722c2b37b9c990e203bbed8a4dece81f2c1a Mon Sep 17 00:00:00 2001 From: Pernekhan Utemuratov Date: Mon, 15 Jul 2024 10:10:26 -0700 Subject: [PATCH] [Misc] Use 0.0.9 version for flashinfer (#6447) Co-authored-by: Pernekhan Utemuratov --- Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 7fbc168ace3dc..89d9be0e84ddb 100644 --- a/Dockerfile +++ b/Dockerfile @@ -172,7 +172,7 @@ RUN --mount=type=bind,from=mamba-builder,src=/usr/src/mamba,target=/usr/src/mamb python3 -m pip install /usr/src/mamba/*.whl --no-cache-dir RUN --mount=type=cache,target=/root/.cache/pip \ - python3 -m pip install https://github.com/flashinfer-ai/flashinfer/releases/download/v0.0.8/flashinfer-0.0.8+cu121torch2.3-cp310-cp310-linux_x86_64.whl + python3 -m pip install https://github.com/flashinfer-ai/flashinfer/releases/download/v0.0.9/flashinfer-0.0.9+cu121torch2.3-cp310-cp310-linux_x86_64.whl #################### vLLM installation IMAGE ####################