This one is AudioLM modified from here https://github.com/lucidrains/audiolm-pytorch to support the music generation needs of Mulan.