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.59k stars 2.09k forks source link

MBD kept loading on each generation #184

Closed GrandaddyShmax closed 1 year ago

GrandaddyShmax commented 1 year ago

probably something that was overlooked but the MBD kept loading on every generation, which was taking more and more VRAM as result.

This change worked for me

adefossez commented 1 year ago

cc @Sparker17 Thanks a lot for the fix @GrandaddyShmax !