Spaces:
Build error
Build error
Update models.py
Browse files
models.py
CHANGED
@@ -107,7 +107,7 @@ class DurationPredictor(nn.Module):
|
|
107 |
|
108 |
self.drop = nn.Dropout(p_dropout)
|
109 |
self.conv_1 = nn.Conv1d(in_channels, filter_channels, kernel_size, padding=kernel_size//2)
|
110 |
-
self.norm_1 = modules.LayerNorm(filter_channels)
|
111 |
self.conv_2 = nn.Conv1d(filter_channels, filter_channels, kernel_size, padding=kernel_size//2)
|
112 |
self.norm_2 = modules.LayerNorm(filter_channels)
|
113 |
self.proj = nn.Conv1d(filter_channels, 1, 1)
|
|
|
107 |
|
108 |
self.drop = nn.Dropout(p_dropout)
|
109 |
self.conv_1 = nn.Conv1d(in_channels, filter_channels, kernel_size, padding=kernel_size//2)
|
110 |
+
self.norm_1 = modules.LayerNorm(filter_channels)
|
111 |
self.conv_2 = nn.Conv1d(filter_channels, filter_channels, kernel_size, padding=kernel_size//2)
|
112 |
self.norm_2 = modules.LayerNorm(filter_channels)
|
113 |
self.proj = nn.Conv1d(filter_channels, 1, 1)
|