NeMo icon indicating copy to clipboard operation
NeMo copied to clipboard

Update Aligner model and tutorial to add NGC checkpoint loading

Open redoctopus opened this issue 2 years ago • 0 comments

Signed-off-by: Jocelyn Huang [email protected]

What does this PR do ?

Adds NGC model loading for the RAD-TTS Aligner.

Collection: TTS

Changelog

  • Add newly published NGC checkpoint to list of pretrained Aligner models
  • Update tutorial to load this checkpoint instead of a local one

Usage

from nemo.collections.tts.models import AlignerModel

aligner = AlignerModel.from_pretrained("tts_en_radtts_aligner")

PR Type:

  • [ ] New Feature
  • [ ] Bugfix
  • [x] Documentation

redoctopus avatar Aug 09 '22 23:08 redoctopus

ran your tutorial a couple times, but still hit error like,

Error(s):	`pynini` not installed, please install via NeMo/nemo_text_processing/pynini_install.sh

See my gist here: https://colab.research.google.com/gist/XuesongYang/918a2ecc7c93cde7c65f3fa0a1241606/aligner_inference_examples.ipynb

@redoctopus, we should add the following to TTS notebooks:

! wget https://raw.githubusercontent.com/NVIDIA/NeMo/main/nemo_text_processing/install_pynini.sh
! bash install_pynini.sh

ekmb avatar Aug 11 '22 10:08 ekmb

@ekmb Thanks!

I've updated the installation in this notebook, and I'll make a PR to update it for all other TTS notebooks shortly.

redoctopus avatar Aug 11 '22 16:08 redoctopus