fix syntax error

mlcommons · Oct 18, 2024 · 783aab4 · 783aab4
1 parent c5c36c2
commit 783aab4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/workload.py b/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/workload.py
@@ -56,7 +56,7 @@ def init_model_fn(
     params = jax_utils.replicate(params)
     return params, model_state
 
-   def model_fn(
+  def model_fn(
       self,
       params: spec.ParameterContainer,
       augmented_and_preprocessed_input_batch: Dict[str, spec.Tensor],