Skip to content

GitLab

Commit 10b6afd6 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Roll back FlashAttention

parent a8e8cfb9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
environment.yml
View file @ 10b6afd6
...@@ -27,5 +27,5 @@ dependencies: ...@@ -27,5 +27,5 @@ dependencies:
- typing-extensions==3.10.0.2 - typing-extensions==3.10.0.2
- pytorch_lightning==1.5.10 - pytorch_lightning==1.5.10
- wandb==0.12.21 - 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 - git+https://github.com/NVIDIA/dllogger.git
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment