From 6fa0d2a2e202171458ed1056e1ff62a907570476 Mon Sep 17 00:00:00 2001
From: Simon Berger <berger@i6.informatik.rwth-aachen.de>
Date: Tue, 5 Nov 2024 16:17:17 +0100
Subject: [PATCH] Add **kwargs to LayerNormNC constructor

---
 i6_models/parts/conformer/norm.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/i6_models/parts/conformer/norm.py b/i6_models/parts/conformer/norm.py
index d0401e55..c46155fe 100644
--- a/i6_models/parts/conformer/norm.py
+++ b/i6_models/parts/conformer/norm.py
@@ -9,11 +9,11 @@ class LayerNormNC(nn.LayerNorm):
     see here: https://pytorch.org/docs/stable/generated/torch.nn.LayerNorm.html
     """
 
-    def __init__(self, channels: int):
+    def __init__(self, channels: int, **kwargs):
         """
         :param channels: number of channels for normalization
         """
-        super().__init__(channels)
+        super().__init__(channels, **kwargs)
 
     def forward(self, tensor: torch.Tensor) -> torch.Tensor:
         """