FastSpeech2
FastSpeech2 copied to clipboard
Minor bug in loading vocoders
These three lines don't load the models into the desired device and should be replaced:
https://github.com/ming024/FastSpeech2/blob/d4e79eb52e8b01d24703b2dfc0385544092958f3/utils/model.py#L20
https://github.com/ming024/FastSpeech2/blob/d4e79eb52e8b01d24703b2dfc0385544092958f3/utils/model.py#L63
https://github.com/ming024/FastSpeech2/blob/d4e79eb52e8b01d24703b2dfc0385544092958f3/utils/model.py#L65
with something like this: torch.load(ckpt_path, map_location=device)
.