transformers
Add Flash Attention 2 to M2M100 model
#30256
Merged

Add Flash Attention 2 to M2M100 model #30256

younesbelkada merged 13 commits into huggingface:main from visheratin:m2m100
visheratin
visheratin Added flash attention 2.
d0ec7dd7
visheratin Fixes.
84020ffb
younesbelkada
younesbelkada commented on 2024-04-16
visheratin
younesbelkada
visheratin Fix inheritance.
5ce57fd6
visheratin Fixed init.
7dc19932
visheratin Remove stuff.
6b863d33
visheratin Added documentation.
17d0797a
visheratin Add FA2 to M2M100 documentation.
2525e1d0
visheratin Add test.
4c4737b1
visheratin
visheratin Fixed documentation.
d1c1d06c
younesbelkada
younesbelkada commented on 2024-04-17
visheratin Update src/transformers/models/m2m_100/modeling_m2m_100.py
b63ed70b
visheratin
HuggingFaceDocBuilderDev
younesbelkada
younesbelkada approved these changes on 2024-04-17
younesbelkada younesbelkada requested a review from amyeroberts amyeroberts 2 years ago
amyeroberts
amyeroberts commented on 2024-04-17
visheratin Update docs/source/en/model_doc/nllb.md
78d75e5d
visheratin Fixed variable name.
f5f02130
visheratin Merge branch 'main' into m2m100
cab70064
visheratin visheratin requested a review from amyeroberts amyeroberts 2 years ago
amyeroberts
amyeroberts approved these changes on 2024-04-17
visheratin
younesbelkada younesbelkada merged b65df514 into main 2 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone