Update: Ignore this repo, check out @lucidrains' implementation https://github.com/lucidrains/musiclm-pytorch