From c000668d8a73520842ed25a6ebfecf85b941aa99 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com>
Date: Sun, 20 Aug 2023 15:13:52 +0800
Subject: [PATCH] Update attentions.py

---
 attentions.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/attentions.py b/attentions.py
index 4a42a1a..ec82ded 100644
--- a/attentions.py
+++ b/attentions.py
@@ -7,7 +7,7 @@ from torch.nn import functional as F
 
 import commons
 import modules
-
+from torch.nn.utils import weight_norm, remove_weight_norm
 class LayerNorm(nn.Module):
   def __init__(self, channels, eps=1e-5):
     super().__init__()