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

app.py

@@ -85,7 +85,7 @@ class InferenceModel(object):
     self.sequence_length = {'inputs': self.inputs_length, 
                             'targets': self.outputs_length}
 
-    self.partitioner = t5x.partitioning.ModelBasedPjitPartitioner(
+    self.partitioner = t5x.partitioning.PjitPartitioner(
         model_parallel_submesh=(1, 1, 1, 1), num_partitions=1)
 
     # Build Codecs and Vocabularies.
@@ -178,7 +178,6 @@ class InferenceModel(object):
     
     Args:
       audio: 1-d numpy array of audio samples (16kHz) for a single example.
-
     Returns:
       A note_sequence of the transcribed audio.
     """