formatting

mlcommons · Oct 17, 2024 · 3913238 · 3913238
1 parent 7ca8365
commit 3913238
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/models.py b/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/models.py
@@ -616,7 +616,12 @@ def __call__(self,
         inputs, input_paddings, train)
 
     inputs = inputs + \
-      ConvolutionBlock(config)(inputs, input_paddings, train, update_batch_norm, use_running_average)
+      ConvolutionBlock(config)(inputs, 
+                               input_paddings, 
+                               train, 
+                               update_batch_norm, 
+                               use_running_average
+                               )
 
     inputs = inputs + 0.5 * FeedForwardModule(config=self.config)(
         inputs, padding_mask, train)