pere commited on
Commit
077c9c4
1 Parent(s): 35e0449
finetune_large_mt5_sentencefix_v4_16.gin CHANGED
@@ -33,7 +33,7 @@ models.EncoderDecoderModel.predict_batch_with_aux.num_decodes = 4
33
 
34
  #partitioning.ModelBasedPjitPartitioner.num_partitions = 2
35
 
36
- partitioning.PjitPartitioner.num_partitions = 2
37
 
38
  #from t5.models import mesh_transformer
39
  #import t5.models
 
33
 
34
  #partitioning.ModelBasedPjitPartitioner.num_partitions = 2
35
 
36
+ partitioning.PjitPartitioner.num_partitions = 1
37
 
38
  #from t5.models import mesh_transformer
39
  #import t5.models