From e441f1b27c9dcb29961e353deb7574a05a424fad 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 14:55:03 +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 03f1dec..95d3527 100644
--- a/attentions.py
+++ b/attentions.py
@@ -31,7 +31,7 @@ class Encoder(nn.Module):
     if isflow:
       cond_layer = torch.nn.Conv1d(kwargs["gin_channels"], 2*hidden_channels*n_layers, 1)
       self.cond_pre = torch.nn.Conv1d(hidden_channels, 2*hidden_channels, 1)
-      self.cond_layer = weight_norm_modules(cond_layer, name='weight')
+      self.cond_layer = weight_norm(cond_layer, name='weight')
       self.gin_channels = kwargs["gin_channels"]
     self.drop = nn.Dropout(p_dropout)
     self.attn_layers = nn.ModuleList()