From ab78ea558ad8af2d443ebd16f3d8f86cea65dcb5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com> Date: Mon, 21 Aug 2023 12:49:20 +0800 Subject: [PATCH] Update attentions.py --- attentions.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/attentions.py b/attentions.py index b757cd2..ecbdbc8 100644 --- a/attentions.py +++ b/attentions.py @@ -68,8 +68,6 @@ 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) attn_mask = x_mask.unsqueeze(2) * x_mask.unsqueeze(-1) x = x * x_mask for i in range(self.n_layers):