From e8e546701086aaf848f8e3582ac9c45f101b6de8 Mon Sep 17 00:00:00 2001 From: Aaron Gokaslan Date: Wed, 13 Mar 2024 11:32:24 -0700 Subject: [PATCH] Update Flash Attention to 2.5.6 --- docker/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 7c9735e13f..24261bb10a 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -276,7 +276,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 && \ - MAX_JOBS=1 pip${PYTHON_VERSION} install --no-cache-dir flash-attn==2.5.0; \ + MAX_JOBS=1 pip${PYTHON_VERSION} install --no-cache-dir flash-attn==2.5.5; \ fi ###############