jaketae commited on
Commit
49a3441
1 Parent(s): 0bb29f1

fix: config should accept kwargs

Browse files
Files changed (1) hide show
  1. 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)