fix: add sampling rate to config
Browse files- configuration_hifigan.py +2 -0
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 |
**kwargs,
|
16 |
):
|
17 |
self.resblock_kernel_sizes = resblock_kernel_sizes
|
@@ -20,4 +21,5 @@ class HiFiGANConfig(PretrainedConfig):
|
|
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)
|
|
|
12 |
upsample_initial_channel=512,
|
13 |
upsample_kernel_sizes=[16, 16, 4, 4],
|
14 |
model_in_dim=80,
|
15 |
+
sampling_rate=22050,
|
16 |
**kwargs,
|
17 |
):
|
18 |
self.resblock_kernel_sizes = resblock_kernel_sizes
|
|
|
21 |
self.model_in_dim = model_in_dim
|
22 |
self.upsample_initial_channel = upsample_initial_channel
|
23 |
self.upsample_kernel_sizes = upsample_kernel_sizes
|
24 |
+
self.sampling_rate = sampling_rate
|
25 |
super().__init__(**kwargs)
|