tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Update attentions.py

Browse Source
This commit is contained in:
Stardust·减
2023-08-20 14:58:15 +08:00
committed by GitHub
parent e441f1b27c
commit 2ad0269e39
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
attentions.py
View File

@@ -43,7 +43,6 @@ class Encoder(nn.Module):
self.norm_layers_1.append(LayerNorm(hidden_channels)) self.norm_layers_1.append(LayerNorm(hidden_channels))
self.ffn_layers.append(FFN(hidden_channels, hidden_channels, filter_channels, kernel_size, p_dropout=p_dropout)) 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)) self.norm_layers_2.append(LayerNorm(hidden_channels))
def forward(self, x, x_mask, g=None): def forward(self, x, x_mask, g=None):
if g is not None: if g is not None:
g = self.cond_layer(g) g = self.cond_layer(g)