transformers
af3de8d8 - [Whisper, Bart, MBart] Add Flash Attention 2 (#27203)

Commit
2 years ago
[Whisper, Bart, MBart] Add Flash Attention 2 (#27203) * add whisper fa2 * correct * change all * correct * correct * fix more * fix more * fix more * fix more * fix more * fix more * Apply suggestions from code review Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> * fix more * fix more * fix more * fix more * fix more --------- Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
Parents
Loading