fits in one line

huggingface · Apr 11, 2023 · e5bd7d8 · e5bd7d8
1 parent f9f87d8
commit e5bd7d8
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/src/transformers/models/cpmant/modeling_cpmant.py b/src/transformers/models/cpmant/modeling_cpmant.py
@@ -47,10 +47,7 @@ class CpmAntLayerNorm(nn.Module):
     We use Root Mean Square (RMS) Layer Normalization, please see https://arxiv.org/abs/1910.07467 for details."
     """
 
-    def __init__(
-        self,
-        config: CpmAntConfig,
-    ):
+    def __init__(self, config: CpmAntConfig,):
         super().__init__()
 
         self.eps = config.eps
@@ -425,10 +422,7 @@ def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
 
 
 class CpmAntSegmentPositionEmbedding(nn.Module):
-    def __init__(
-        self,
-        config: CpmAntConfig,
-    ):
+    def __init__(self, config: CpmAntConfig):
         super().__init__()
 
         self.num_heads = config.num_attention_heads