From 8081ab015262b8465747e317c8e8e63eab0735d3 Mon Sep 17 00:00:00 2001 From: lwj-st <109193776+lwj-st@users.noreply.github.com> Date: Wed, 25 Sep 2024 11:54:19 +0800 Subject: [PATCH] Update Dockerfile_tag --- Dockerfile_tag | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile_tag b/Dockerfile_tag index f394044..4ca2ebd 100644 --- a/Dockerfile_tag +++ b/Dockerfile_tag @@ -50,7 +50,7 @@ RUN bash -c "source activate lazyllm && conda install -y mpi4py" \ && bash -c "source activate lazyllm && pip install -r requirements1.txt --default-timeout=10000 --no-deps --no-cache-dir " \ && bash -c "source activate lazyllm && pip install -r requirements2.txt --default-timeout=10000 --no-deps --no-cache-dir " \ && bash -c "source activate lazyllm && pip install -r requirements3.txt --default-timeout=10000 --no-deps --no-cache-dir " \ - && bash -c "source activate lazyllm && pip install flash-attn==2.6.3 && pip cache purge" \ + && bash -c "source activate lazyllm && pip install flash-attn==2.6.2 && pip cache purge" \ && rm -rf /tmp/* ARG LAZYLLM_VERSION=""