Replace deprecated `t5x.partitioning.ModelBasedPjitPartitioner` with `t5x.partitioning.PjitPartitioner`
8a60e51
vilimus
commited on