transformers
a56a23f3 - [utils] Add MUSA support for Flash Attention 2 (#48612)

Commit
8 days ago
[utils] Add MUSA support for Flash Attention 2 (#48612) * [utils] Add MUSA support for Flash Attention 2 * Update src/transformers/utils/import_utils.py simplify logic Co-authored-by: Anton Vlasjuk <73884904+vasqu@users.noreply.github.com> * [utils] Address MUSA Flash Attention review feedback * [utils] Format Flash Attention availability check * [utils] Remove redundant MUSA Flash Attention test --------- Co-authored-by: Anton Vlasjuk <73884904+vasqu@users.noreply.github.com>
Parents
Loading