dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
transformers/tests/models/musicgen_melody
History
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 Melody (#28819) 2024-03-18 13:06:12 +00:00
test_feature_extraction_musicgen_melody.py Add MusicGen Melody (#28819) 2024-03-18 13:06:12 +00:00
test_modeling_musicgen_melody.py Add Flash Attention 2 support to Musicgen and Musicgen Melody (#29939) 2024-04-02 11:23:49 +01:00
test_processor_musicgen_melody.py Add MusicGen Melody (#28819) 2024-03-18 13:06:12 +00:00