Add Flash Attention 2 to M2M100 model #30256
Added flash attention 2.
d0ec7dd7
Fixes.
84020ffb
Fix inheritance.
5ce57fd6
Fixed init.
7dc19932
Remove stuff.
6b863d33
Added documentation.
17d0797a
Add FA2 to M2M100 documentation.
2525e1d0
Add test.
4c4737b1
Fixed documentation.
d1c1d06c
Update src/transformers/models/m2m_100/modeling_m2m_100.py
b63ed70b
Update docs/source/en/model_doc/nllb.md
78d75e5d
Fixed variable name.
f5f02130
Merge branch 'main' into m2m100
cab70064
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub