rename checkpoint filename

pytorch · Jan 5, 2023 · e2c75a4 · e2c75a4
1 parent 068be61
commit e2c75a4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/torchaudio/prototype/pipelines/hifigan_pipeline.py b/torchaudio/prototype/pipelines/hifigan_pipeline.py
@@ -35,7 +35,7 @@ class HiFiGANVocoderBundle:
         >>>
         >>> # Load the HiFiGAN bundle
         >>> vocoder = bundle.get_vocoder()
-        Downloading: "https://download.pytorch.org/torchaudio/models/hifigan_generator_v3_ljspeech.pth"
+        Downloading: "https://download.pytorch.org/torchaudio/models/hifigan_vocoder_v3_ljspeech.pth"
         100%|████████████| 5.59M/5.59M [00:00<00:00, 18.7MB/s]
         >>>
         >>> # Generate synthetic mel spectrogram
@@ -63,7 +63,7 @@ class HiFiGANVocoderBundle:
         >>>
         >>> # Load HiFiGAN bundle
         >>> vocoder = bundle_hifigan.get_vocoder()
-        Downloading: "https://download.pytorch.org/torchaudio/models/hifigan_generator_v3_ljspeech.pth"
+        Downloading: "https://download.pytorch.org/torchaudio/models/hifigan_vocoder_v3_ljspeech.pth"
         100%|████████████| 5.59M/5.59M [00:03<00:00, 1.55MB/s]
         >>>
         >>> # Use HiFiGAN to convert mel spectrogram to audio
@@ -186,7 +186,7 @@ def forward(self, waveform: torch.Tensor) -> torch.Tensor:
 
 
 HIFIGAN_VOCODER_V3_LJSPEECH = HiFiGANVocoderBundle(
-    "hifigan_generator_v3_ljspeech.pth",
+    "hifigan_vocoder_v3_ljspeech.pth",
     _vocoder_params={
         "upsample_rates": (8, 8, 4),
         "upsample_kernel_sizes": (16, 16, 8),