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.18k stars 2.01k forks source link

remove compression_model from lm checkpoints #368

Closed adefossez closed 6 months ago

adefossez commented 6 months ago

The compression_model is stored inside the checkpoint, but not reliably used in different parts of the code. For consistency, it should always be loaded from the origina compression_model_checkpoint and thus is not needed in the checkpoint file.