bump flash attention to 2.6.2 (#1781) [skip ci]
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user