Imported from https://github.com/lokhiufung/music_generation