Also made a pull request and accepted CLA
https://github.com/facebookresearch/audiocraft/pull/17
Thank you so much for keeping this open