Add Flash Attention 2 support to Musicgen and Musicgen Melody#29939

Merged

ylacombe merged 16 commits intohuggingface:mainfrom

ylacombe:add-FA2-musicgen

Apr 2, 2024

Commits on Mar 29, 2024

Commits on Apr 1, 2024

refactor attention implementation arguments
ylacombe
committed