Update attentions.py
This commit is contained in:
@@ -43,7 +43,6 @@ class Encoder(nn.Module):
|
||||
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.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)
|
||||
|
||||
Reference in New Issue
Block a user