From 172d75134e2a8be12e4277a3056dbe224aae6ff1 Mon Sep 17 00:00:00 2001
From: Woosung Choi <Woosung.Choi@sony.com>
Date: Mon, 1 May 2023 10:17:30 +0900
Subject: [PATCH] Update norm.py

---
 encodec/modules/norm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/encodec/modules/norm.py b/encodec/modules/norm.py
index 19970e0..8ec476d 100644
--- a/encodec/modules/norm.py
+++ b/encodec/modules/norm.py
@@ -25,4 +25,4 @@ def forward(self, x):
         x = einops.rearrange(x, 'b ... t -> b t ...')
         x = super().forward(x)
         x = einops.rearrange(x, 'b t ... -> b ... t')
-        return
+        return x