Skip to content

Commit e499848

Browse files
committed
fix the bash in docker base
1 parent 0484f64 commit e499848

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

docker/Dockerfile-base

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,9 +40,11 @@ RUN git lfs install --skip-repo && \
4040

4141
RUN if [ "$TORCH_CUDA_ARCH_LIST" = "9.0+PTX" ] ; then \
4242
git clone https://github.com/Dao-AILab/flash-attention.git; \
43+
cd flash-attention; \
4344
git checkout v2.7.4.post1; \
44-
cd flash-attention/hopper; \
45+
cd hopper; \
4546
FLASH_ATTENTION_DISABLE_SM80=TRUE FLASH_ATTENTION_DISABLE_FP8=TRUE MAX_JOBS=128 python setup.py install; \
46-
elif if [ "$PYTORCH_VERSION" = "2.7.0" ] ; then \
47+
cd ../..; \
48+
elif[ "$PYTORCH_VERSION" = "2.7.0" ] ; then \
4749
pip3 install flash-attn==2.7.4.post1; \
4850
fi

0 commit comments

Comments
 (0)