Replace deprecated `t5x.partitioning.ModelBasedPjitPartitioner` with `t5x.partitioning.PjitPartitioner`

#2
by deleted - opened
Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment