newer 'generating piano music with transformers' colab?

[Michael Theodore]

Hello, I'm wondering if some of the notebooks from here: https://magenta.tensorflow.org/demos/colab/ have been updated.

For instance, Generating Piano Music with Transformers doesn't seem to work on colab anymore (tensorflow 1 no longer supported).

Is there a newer version?

thanks!, Michael