Open Amazing-Ming opened 1 year ago
update jax and t5x,(may need to uninstall package [jestimator] first)
I also encountered this problem,能不能说的更详细一点
It's renamed in new version jax to jax.experimental.array_serialization
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' {} +
No module named 'jax.experimental.gda_serialization'