fix: config should accept kwargs
Browse files- configuration_hifigan.py +2 -1
configuration_hifigan.py
CHANGED
@@ -12,6 +12,7 @@ class HiFiGANConfig(PretrainedConfig):
|
|
12 |
upsample_initial_channel=512,
|
13 |
upsample_kernel_sizes=[16, 16, 4, 4],
|
14 |
model_in_dim=80,
|
|
|
15 |
):
|
16 |
self.resblock_kernel_sizes = resblock_kernel_sizes
|
17 |
self.resblock_dilation_sizes = resblock_dilation_sizes
|
@@ -19,4 +20,4 @@ class HiFiGANConfig(PretrainedConfig):
|
|
19 |
self.model_in_dim = model_in_dim
|
20 |
self.upsample_initial_channel = upsample_initial_channel
|
21 |
self.upsample_kernel_sizes = upsample_kernel_sizes
|
22 |
-
super().__init__()
|
|
|
12 |
upsample_initial_channel=512,
|
13 |
upsample_kernel_sizes=[16, 16, 4, 4],
|
14 |
model_in_dim=80,
|
15 |
+
**kwargs,
|
16 |
):
|
17 |
self.resblock_kernel_sizes = resblock_kernel_sizes
|
18 |
self.resblock_dilation_sizes = resblock_dilation_sizes
|
|
|
20 |
self.model_in_dim = model_in_dim
|
21 |
self.upsample_initial_channel = upsample_initial_channel
|
22 |
self.upsample_kernel_sizes = upsample_kernel_sizes
|
23 |
+
super().__init__(**kwargs)
|