Update app.py
Browse files
app.py
CHANGED
@@ -97,7 +97,7 @@ class InferenceModel(object):
|
|
97 |
'targets': self.outputs_length}
|
98 |
|
99 |
self.partitioner = t5x.partitioning.PjitPartitioner(
|
100 |
-
model_parallel_submesh=
|
101 |
|
102 |
# Build Codecs and Vocabularies.
|
103 |
self.spectrogram_config = spectrograms.SpectrogramConfig()
|
|
|
97 |
'targets': self.outputs_length}
|
98 |
|
99 |
self.partitioner = t5x.partitioning.PjitPartitioner(
|
100 |
+
model_parallel_submesh=none, num_partitions=1)
|
101 |
|
102 |
# Build Codecs and Vocabularies.
|
103 |
self.spectrogram_config = spectrograms.SpectrogramConfig()
|