Alexander Visheratin b65df514d1 Add Flash Attention 2 to M2M100 model (#30256) ... * Added flash attention 2. * Fixes. * Fix inheritance. * Fixed init. * Remove stuff. * Added documentation. * Add FA2 to M2M100 documentation. * Add test. * Fixed documentation. * Update src/transformers/models/m2m_100/modeling_m2m_100.py Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com> * Update docs/source/en/model_doc/nllb.md Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> * Fixed variable name. --------- Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com> Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>		2024-04-18 10:27:58 +02:00
..
__init__.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
test_modeling_m2m_100.py	Add Flash Attention 2 to M2M100 model (#30256)	2024-04-18 10:27:58 +02:00
test_tokenization_m2m_100.py	Remove static pretrained maps from the library's internals (#29112)	2024-03-25 10:33:38 +01:00