cse556-music-transformer-with-difficulty