facebookresearch / audiocraft

Audiocraft is a library for audio processing and generation with deep learning. It features the state-of-the-art EnCodec audio compressor / tokenizer, along with MusicGen, a simple and controllable music generation LM with textual and melodic conditioning.
MIT License
20.15k stars 2.01k forks source link

Fix FSDP support with pytorch 2.1.0 #454

Closed JadeCopet closed 2 months ago

JadeCopet commented 2 months ago

Fix FSDP code wrote for torch 2.0 but breaking for 2.1.0 as raised by https://github.com/facebookresearch/audiocraft/issues/358 and https://github.com/facebookresearch/audiocraft/issues/450 .