whatmodel.sh / tts / Tacotron 2 + WaveGlow (NVIDIA)
Tacotron 2 + WaveGlow (NVIDIA)
↗ share
+ contribute
Tacotron 2 + WaveGlow (NVIDIA) is an open-weights text-to-speech model from NVIDIA, released 2018. Licence: bsd-3-clause.
Catalogue identity notes
Exact NVIDIA implementation with WaveGlow vocoder. Do not transfer these listening-study values to Coqui, SpeechBrain, HiFi-GAN or a differently trained checkpoint. Historical run dependencies documented upstream; no current one-command run verified.
Model information sources
$ how it compares
Scored text-to-speech models released closest to Tacotron 2 + WaveGlow (NVIDIA), out of 148 with a score, then more from NVIDIA.
| # | model | score | where | ||
|---|---|---|---|---|---|
| 81 | 50±16 | comparison only | |||
| 82 | 50±16 | comparison only | |||
| 124 | 34±5 | local | |||
| 135 | 28±6 | comparison only | |||
| 142 | 19±6 | comparison only | |||
| more from NVIDIA | |||||
| 113 | 39±6 | local | |||
see all text-to-speech models · find the ones that fit your machine
$ what it's good at
other tests1 not in a task · 0.0% of its score
| test | result | vs the others in its run | counts |
|---|---|---|---|
| FastPitch 2020 LJ Speech human MOS lab | 3.95 / 5 ✎ | █··········█ beats 0/1 | — |
$ where to run it
Other ways to run it
- Run it as published · needs ~1.19 GB, NVIDIA only · from github.com/pytorch
pip install torch numpy scipy librosa unidecode inflect && apt-get update && apt-get install -y libsndfile1 && python -c "import torch; from scipy.io.wavfile import write; tacotron2 = torch.hub.load('NVIDIA/DeepLearningExamples:torchhub', 'nvidia_tacotron2', model_math='fp16', trust_repo=True); tacotron2 = tacotron2.to('cuda'); tacotron2.eval(); waveglow = torch.hub.load('NVIDIA/DeepLearningExamples:torchhub', 'nvidia_waveglow', model_math='fp16', trust_repo=True); waveglow = waveglow.remove_weightnorm(waveglow); waveglow = waveglow.to('cuda'); waveglow.eval(); utils = torch.hub.load('NVIDIA/DeepLearningExamples:torchhub', 'nvidia_tts_utils', trust_repo=True); sequences, lengths = utils.prepare_input_sequence(['Hello world, I missed you so much.']); torch.set_grad_enabled(False); mel, _, _ = tacotron2.infer(sequences, lengths); audio = waveglow.infer(mel); write('audio.wav', 22050, audio[0].data.cpu().numpy())"
Official weights and documentation: github.com/NVIDIA ↗
$ published results · 1 result, not enough overlap for a score
| test | result | source and conditions |
|---|---|---|
| FastPitch 2020 LJ Speech human MOS | 3.95 / 5 | NVIDIA FastPitch: original listening studies in committed README · lab-reported Committed README MOS table: 3.946 ± 0.134 (95% CI); grapheme inputs and WaveGlow (FastPitch paper Table 1). Store (MOS−1)/4. |
$ improve this page
Use a short GitHub form or edit the data through a pull request. Every number needs a source; here's which ones we can use.
data updated · sources and licences · json