Roll back FlashAttention

10b6afd6 · Gustaf Ahdritz · a8e8cfb9 · 10b6afd6
Commit 10b6afd6 authored Jul 11, 2022 by Gustaf Ahdritz
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

environment.yml environment.yml +1 -1

No files found.
--- a/environment.yml
+++ b/environment.yml
@@ -27,5 +27,5 @@ dependencies:
      - typing-extensions==3.10.0.2
      - pytorch_lightning==1.5.10
      - wandb==0.12.21
-      - git+https://github.com/HazyResearch/flash-attention.git
+      - git+https://github.com/HazyResearch/flash-attention.git@5b838a8bef78186196244a4156ec35bbb58c337d
      - git+https://github.com/NVIDIA/dllogger.git