upgrade to flash-attn 2.7.0 (#2048)

This commit is contained in:
Wing Lian
2024-11-14 06:59:25 -05:00
committed by GitHub
parent 5e98cdddac
commit 2d7830fda6
4 changed files with 3 additions and 7 deletions

View File

@@ -12,7 +12,7 @@ addict
fire
PyYAML>=6.0
requests
flash-attn==2.6.3
flash-attn==2.7.0.post2
sentencepiece
wandb
einops