Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

tts_waveglow_268m from_pretrained KeyError #256

Open
markwny opened this issue Jul 5, 2021 · 0 comments
Open

tts_waveglow_268m from_pretrained KeyError #256

markwny opened this issue Jul 5, 2021 · 0 comments

Comments

@markwny
Copy link

markwny commented Jul 5, 2021

Error received when trying to load pretrained waveglow model.
vocoder = WaveGlowModel.from_pretrained(model_name="tts_waveglow_268m")

Error received:

`~/.virtualenvs/nemo/lib/python3.8/site-packages/nemo/collections/tts/models/waveglow.py in load_state_dict(self, state_dict, strict)
257 # Ideally, we should remove this during saving instead of ignoring during loading
258 for i in range(self._cfg.waveglow.n_flows):
--> 259 del state_dict[f"waveglow.convinv.{i}.inv_conv.weight"]
260 super().load_state_dict(state_dict, strict=strict)

KeyError: 'waveglow.convinv.0.inv_conv.weight'`

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant