jaketae commited on
Commit
ea79692
1 Parent(s): ce4b344

fix: add sampling rate to config

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