Fix yapf

mlcommons · Dec 12, 2023 · c5e9a03 · c5e9a03
1 parent a026ce7
commit c5e9a03
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/algorithmic_efficiency/workloads/wmt/wmt_pytorch/models.py b/algorithmic_efficiency/workloads/wmt/wmt_pytorch/models.py
@@ -578,8 +578,7 @@ def __init__(self,
             pre_ln=pre_ln) for _ in range(num_layers)
     ])
     self.num_layers = num_layers
-    self.norm = (
-        nn.LayerNorm(d_model, eps=layer_norm_eps) if pre_ln else None)
+    self.norm = (nn.LayerNorm(d_model, eps=layer_norm_eps) if pre_ln else None)
 
   def forward(self,
               tgt: Tensor,