From f0d9093e7240447491114e0b9c892189d6ba983b Mon Sep 17 00:00:00 2001 From: Saaketh Narayan Date: Mon, 29 Apr 2024 17:07:38 -0700 Subject: [PATCH] use fa2 258, no build iso (#3224) --- docker/Dockerfile | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 79fc4e2842..c147fe45a7 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -283,9 +283,7 @@ RUN if [ -n "$MOFED_VERSION" ] ; then \ RUN if [ -n "$CUDA_VERSION" ] ; then \ pip${PYTHON_VERSION} install --upgrade --no-cache-dir ninja==1.11.1 && \ pip${PYTHON_VERSION} install --upgrade --no-cache-dir --force-reinstall packaging==22.0 && \ - git clone --branch v2.5.7 https://github.com/Dao-AILab/flash-attention.git && \ - cd flash-attention && \ - MAX_JOBS=1 python${PYTHON_VERSION} setup.py install && \ + MAX_JOBS=1 pip${PYTHON_VERSION} install --no-cache-dir --no-build-isolation flash-attn==2.5.8; \ cd .. ; \ fi