Yoach Lacombe 0d04b1e25a Add Flash Attention 2 support to Musicgen and Musicgen Melody (#29939) ... * add FA2 to o.g Musicgen * make style * add FA2 support to Musicgen Melody * add generation FA2 tests to o.g Musicgen * make style and fix copies * add Musicgen to FA2 docs + deprecate list * add sdpa supports to Musicgen's * make style and fix copies * refactor attention implementation arguments * add Copied from to sdpa tests * add copied form in sdpa tests melody * add copied for FA2 generation tests * add FA2 inference copied from * make style		2024-04-02 11:23:49 +01:00
..
__init__.py	Add Musicgen (#24109)	2023-06-29 14:48:59 +01:00
test_modeling_musicgen.py	Add Flash Attention 2 support to Musicgen and Musicgen Melody (#29939)	2024-04-02 11:23:49 +01:00
test_processing_musicgen.py	Add # Copied from statements to audio feature extractors that use the floats_list function (#26581)	2023-10-04 17:09:48 +02:00