[`FA2`] Add flash attention for opt #26414
amyeroberts
merged
16 commits into
huggingface:main
from
susnato:flash_attn_opt
added flash attention for opt
626276f1
younesbelkada
commented
on 2023-09-26
younesbelkada
commented
on 2023-09-26
added to list
7800457d
fix use cache (#3)
689f5991
younesbelkada
commented
on 2023-09-26
style fix
74e96876
younesbelkada
commented
on 2023-09-26
fix text
cf923e88
younesbelkada
commented
on 2023-09-26
test fix2
db8cf07e
younesbelkada
commented
on 2023-09-26
reverted until 689f599
af67e0f5
torch fx tests are working now!
edf1610b
small fix
f34b680a
younesbelkada
commented
on 2023-09-28
Merge branch 'main' into flash_attn_opt
90be2103
added TODO docstring
5cec2adb
Merge branch 'main' into flash_attn_opt
2bde7cf1
Merge branch 'main' into flash_attn_opt
db18d65f
changes
adfbb69c
susnato
changed the title
younesbelkada
commented
on 2023-11-07
Merge branch 'main' into flash_attn_opt
10ab9b30
comments and .md file modification
7d4c688a
amyeroberts
approved these changes
on 2023-11-21
younesbelkada
approved these changes
on 2023-11-22
younesbelkada
approved these changes
on 2023-11-22
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub