blakelemoine/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

map vocoder into cpu

Browse Source
This commit is contained in:
James Betker 2022-03-21 17:10:57 -06:00
parent 9e97cd800c
commit 3692c4cae3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/scripts/audio/gen/speech_synthesis_utils.py
View File

@ -22,7 +22,7 @@ def load_speech_dvae():
def load_univnet_vocoder(): def load_univnet_vocoder():
model = UnivNetGenerator() model = UnivNetGenerator()
sd = torch.load('../experiments/univnet_c32_pretrained_libri.pt') sd = torch.load('../experiments/univnet_c32_pretrained_libri.pt', map_location='cpu')
model.load_state_dict(sd['model_g']) model.load_state_dict(sd['model_g'])
model = model.cpu() model = model.cpu()
model.eval(inference=True) model.eval(inference=True)