FastSpeech2 icon indicating copy to clipboard operation
FastSpeech2 copied to clipboard

Minor bug in loading vocoders

Open Mahyar24 opened this issue 1 year ago • 0 comments

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).

Mahyar24 avatar Dec 27 '23 18:12 Mahyar24