From 64f6abe127717fa7ffed2036110894e02fd0f93e 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:58:36 +0800
Subject: [PATCH] Update attentions.py

---
 attentions.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/attentions.py b/attentions.py
index b6804d4..d752f24 100644
--- a/attentions.py
+++ b/attentions.py
@@ -44,8 +44,8 @@ class Encoder(nn.Module):
       self.ffn_layers.append(FFN(hidden_channels, hidden_channels, filter_channels, kernel_size, p_dropout=p_dropout))
       self.norm_layers_2.append(LayerNorm(hidden_channels))
    def forward(self, x, x_mask, g=None):
-       if g is not None:
-           g = self.cond_layer(g)
+    if g is not None:
+      g = self.cond_layer(g)
     attn_mask = x_mask.unsqueeze(2) * x_mask.unsqueeze(-1)
     x = x * x_mask
     for i in range(self.n_layers):