use fast group norm

2025-09-01 04:14:38 +08:00 · 2024-09-15 08:47:10 -07:00
parent c3209fd29a
commit 0a73862430
1 changed files with 1 additions and 1 deletions
--- a/encodec/encodec.py
+++ b/encodec/encodec.py
@@ -30,7 +30,7 @@ class EncodecConv1d(nn.Module):
            in_channels, out_channels, kernel_size, stride, dilation=dilation
        )
        if self.norm_type == "time_group_norm":
-            self.norm = nn.GroupNorm(1, out_channels)
+            self.norm = nn.GroupNorm(1, out_channels, pytorch_compatible=True)

        self.stride = stride