Merge pull request #360 from chuanli11/fix/dockerfile

remove checkout v2.0.0.post1 from dockerfile

Merge pull request #360 from chuanli11/fix/dockerfile
remove checkout v2.0.0.post1 from dockerfile
2dd87d06 · Tri Dao · GitHub · b8020d73 · 30fd8c17 · 2dd87d06
Unverified Commit 2dd87d06 authored Jul 20, 2023 by Tri Dao Committed by GitHub Jul 20, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

training/Dockerfile training/Dockerfile +1 -1

No files found.
--- a/training/Dockerfile
+++ b/training/Dockerfile
@@ -89,7 +89,7 @@ RUN pip install flash-attn==2.0.0.post1

 # Install CUDA extensions for cross-entropy, fused dense, layer norm
 RUN git clone https://github.com/HazyResearch/flash-attention \
-    && cd flash-attention && git checkout v2.0.0.post1 \
+    && cd flash-attention \
    && cd csrc/fused_softmax && pip install . && cd ../../ \
    && cd csrc/rotary && pip install . && cd ../../ \
    && cd csrc/xentropy && pip install . && cd ../../ \