bump flash attention to 2.6.2 (#1781) [skip ci]

This commit is contained in:
Wing Lian
2024-07-23 19:54:15 -04:00
committed by GitHub
parent 608a2f3180
commit e6b299dd79
2 changed files with 3 additions and 3 deletions

View File

@@ -12,7 +12,7 @@ fire
PyYAML>=6.0
requests
datasets==2.19.1
flash-attn==2.6.1
flash-attn==2.6.2
sentencepiece
wandb
einops