remove extra layer norms (ecmwf#1589)

Author: kctezcan

src/weathergen/model/engines.py

@@ -446,14 +446,6 @@ def __init__(self, cf: Config, num_healpix_cells: int, dim_aux: int = None) -> N
                         torch.nn.LayerNorm(self.cf.ae_global_dim_embed, elementwise_affine=False)
                     )
 
-            self.fe_blocks.append(
-                torch.nn.LayerNorm(self.cf.ae_global_dim_embed, elementwise_affine=False)
-            )
-
-            self.fe_blocks.append(
-                torch.nn.LayerNorm(self.cf.ae_global_dim_embed, elementwise_affine=False)
-            )
-
         def init_weights_final(m):
             if isinstance(m, torch.nn.Linear):
                 torch.nn.init.normal_(m.weight, mean=0, std=0.001)