Skip to content

Commit

Permalink
fix bug in tutorial (NVIDIA#3546)
Browse files Browse the repository at this point in the history
Signed-off-by: Oktai Tatanov <[email protected]>
  • Loading branch information
Oktai15 authored Jan 28, 2022
1 parent 5484652 commit 7fbddc4
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion tutorials/tts/Inference_ModelSelect.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -159,6 +159,9 @@
"\n",
"def load_spectrogram_model():\n",
" override_conf = None\n",
" \n",
" from_pretrained_call = SpectrogramGenerator.from_pretrained\n",
" \n",
" if spectrogram_generator == \"tacotron2\":\n",
" from nemo.collections.tts.models import Tacotron2Model\n",
" pretrained_model = \"tts_en_tacotron2\"\n",
Expand All @@ -181,6 +184,7 @@
" elif spectrogram_generator == \"talknet\":\n",
" from nemo.collections.tts.models import TalkNetSpectModel\n",
" pretrained_model = \"tts_en_talknet\"\n",
" from_pretrained_call = TalkNetSpectModel.from_pretrained\n",
" elif spectrogram_generator == \"fastpitch\":\n",
" from nemo.collections.tts.models import FastPitchModel\n",
" pretrained_model = \"tts_en_fastpitch\"\n",
Expand All @@ -196,7 +200,7 @@
" else:\n",
" raise NotImplementedError\n",
" \n",
" model = SpectrogramGenerator.from_pretrained(pretrained_model, override_config_path=override_conf)\n",
" model = from_pretrained_call(pretrained_model, override_config_path=override_conf)\n",
" \n",
" return model\n",
"\n",
Expand Down

0 comments on commit 7fbddc4

Please sign in to comment.