NeMo
NeMo copied to clipboard
Update Aligner model and tutorial to add NGC checkpoint loading
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
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 Thanks!
I've updated the installation in this notebook, and I'll make a PR to update it for all other TTS notebooks shortly.