diff --git a/hifigan/hub/__init__.py b/hifigan/hub/__init__.py index d9f10bb..85205f3 100644 --- a/hifigan/hub/__init__.py +++ b/hifigan/hub/__init__.py @@ -9,7 +9,7 @@ def hifigan_48k( progress: bool = True, ) -> Generator: hifigan = Generator( - initial_channel=256, + initial_channel=128, resblock="1", resblock_kernel_sizes=[3,7,11], resblock_dilation_sizes=[ diff --git a/test.py b/test.py index b990595..85b0408 100644 --- a/test.py +++ b/test.py @@ -35,7 +35,7 @@ def load_remote(): wav, sr = torchaudio.load("zszy_48k.wav") assert sr == 48000 -mel = mel_spectrogram_torch(wav, 2048, 256, 48000, 512, 2048, 0, None, False) +mel = mel_spectrogram_torch(wav, 2048, 128, 48000, 512, 2048, 0, None, False) mel = mel.to(device) out = hifigan(mel)