From 4ef2e8293fd2b48517b543445b2bcfabca5b6ce2 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing@axolotl.ai>
Date: Fri, 16 May 2025 15:45:52 -0400
Subject: [PATCH] fix the bash in docker base

---
 docker/Dockerfile-base | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/docker/Dockerfile-base b/docker/Dockerfile-base
index ccb87456a..a35110ef5 100644
--- a/docker/Dockerfile-base
+++ b/docker/Dockerfile-base
@@ -40,9 +40,11 @@ RUN git lfs install --skip-repo && \
 
 RUN if [ "$TORCH_CUDA_ARCH_LIST" = "9.0+PTX" ] ; then \
         git clone https://github.com/Dao-AILab/flash-attention.git; \
+        cd flash-attention; \
         git checkout v2.7.4.post1; \
-        cd flash-attention/hopper; \
+        cd hopper; \
         FLASH_ATTENTION_DISABLE_SM80=TRUE FLASH_ATTENTION_DISABLE_FP8=TRUE MAX_JOBS=128 python setup.py install; \
-    elif if [ "$PYTORCH_VERSION" = "2.7.0" ] ; then \
+        cd ../..; \
+    elif[ "$PYTORCH_VERSION" = "2.7.0" ] ; then \
         pip3 install flash-attn==2.7.4.post1; \
     fi