search

magenta / mt3

MT3: Multi-Task Multitrack Music Transcription
Apache License 2.0
1.43k stars 187 forks source link

AttributeError: module 'jax' has no attribute 'tree_multimap' #71

Closed Quandong-Zhang closed 2 years ago

Quandong-Zhang commented 2 years ago

image

The jax package doesn't seem to have the correct version installed. This version of the jax package does not seem to have a function called 'tree_multimap'

Quandong-Zhang commented 2 years ago

I found out that the jax version in Colab is 0.3.16. But the jax package version required for this project to run is 0.3.15.

trevorwelch commented 2 years ago

If you're looking to use the existing Colab, I fixed this by adding:

!pip install jax==0.3.15
!pip install jaxlib==0.3.15

just below the !gsutil -q -m cp gs://magentadata/soundfonts/SGM-v2.01-Sal-Guit-Bass-V1.3.sf2 . line in the Setup Environment cell.

iansimon commented 2 years ago

This is going to be difficult to resolve properly until colab supports Python versions more recent than 3.7, but #74 should be a temporary workaround as suggested by @Quandong-Zhang and @trevorwelch above.