storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-17 00:23:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix vocoder paths

Browse Source
This commit is contained in:
Brandon Thomas
2023-03-21 05:53:07 -04:00
parent fc27fb26c7
commit 8bac589838
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tts/tacotron2_v1_early_fakeyou/vocodes_inference_updated.py
+3 -3
View File
@@ -78,10 +78,10 @@ pipeline = FakeYouTt2Pipeline()
# NB: Not sure if there was a historical decision to load the models first outside of the pipeline.
# Probably not, though. Seems like this could be cleaned up.
if args.vocoder_type == 'hifigan-superres':
pipeline.maybe_load_waveglow_model(args.waveglow_vocoder_checkpoint_path)
pipeline.maybe_load_hifigan(args.hifigan_vocoder_checkpoint_path)
pipeline.maybe_load_hifigan_super_resolution(args.hifigan_superres_vocoder_checkpoint_path)
elif args.vocoder_type == 'waveglow':
pipeline.maybe_load_waveglow_model(args.hifigan_vocoder_checkpoint_path)
pipeline.maybe_load_waveglow_model(args.hifigan_superres_vocoder_checkpoint_path)
pipeline.maybe_load_waveglow_model(args.waveglow_vocoder_checkpoint_path)
else:
raise Exception('Wrong vocoder type: {}'.format(args.vocoder_type))