search

magenta / mt3

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

No module named 'jax.experimental.gda_serialization' #129

Open Amazing-Ming opened 1 year ago

Amazing-Ming commented 1 year ago

No module named 'jax.experimental.gda_serialization'

Chunyuan-Li commented 1 year ago

update jax and t5x，(may need to uninstall package [jestimator] first)

lvxiangyi commented 1 year ago

I also encountered this problem，能不能说的更详细一点

tikhomirovd commented 1 year ago

It's renamed in new version jax to jax.experimental.array_serialization

shuntacurosu commented 1 year ago

This issue can be temporarily resolved by renaming the package name. To do this, run the following block immediately before the "Imports and Definitions" section.

#@title Replace package name for t5x
!find /usr/local/lib/python3.10/dist-packages/orbax -type f -name "*.py" -exec sed -i 's/gda_serialization/array_serialization/g' {} +