reformat HybridLayer __init__

Zyphra · Oct 4, 2024 · 0a90fc7 · 0a90fc7
1 parent 267530d
commit 0a90fc7
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/transformers/models/zamba/modeling_zamba.py b/src/transformers/models/zamba/modeling_zamba.py
@@ -978,9 +978,7 @@ def forward(
 
 
 class HybridLayer(nn.Module):
-    def __init__(
-        self, shared_transf: ZambaAttentionDecoderLayer, linear: nn.Linear, mamba: ZambaMambaDecoderLayer
-    ):
+    def __init__(self, shared_transf: ZambaAttentionDecoderLayer, linear: nn.Linear, mamba: ZambaMambaDecoderLayer):
         super().__init__()
         self.shared_transf = shared_transf
         self.linear = linear