|
@@ -183,7 +183,7 @@ class ResBlock1(torch.nn.Module):
|
|
|
remove_parametrizations(conv, tensor_name="weight")
|
|
remove_parametrizations(conv, tensor_name="weight")
|
|
|
|
|
|
|
|
|
|
|
|
|
-class ParralelBlock(nn.Module):
|
|
|
|
|
|
|
+class ParallelBlock(nn.Module):
|
|
|
def __init__(
|
|
def __init__(
|
|
|
self,
|
|
self,
|
|
|
channels: int,
|
|
channels: int,
|
|
@@ -254,7 +254,7 @@ class HiFiGANGenerator(nn.Module):
|
|
|
for i in range(len(self.ups)):
|
|
for i in range(len(self.ups)):
|
|
|
ch = upsample_initial_channel // (2 ** (i + 1))
|
|
ch = upsample_initial_channel // (2 ** (i + 1))
|
|
|
self.resblocks.append(
|
|
self.resblocks.append(
|
|
|
- ParralelBlock(ch, resblock_kernel_sizes, resblock_dilation_sizes)
|
|
|
|
|
|
|
+ ParallelBlock(ch, resblock_kernel_sizes, resblock_dilation_sizes)
|
|
|
)
|
|
)
|
|
|
|
|
|
|
|
self.activation_post = post_activation()
|
|
self.activation_post = post_activation()
|
Lengyue