[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>