Load checkpoints on CPU

app.py

@@ -188,13 +188,13 @@ def main():
     net_g = SynthesizerTrn(hps.data.filter_length // 2 + 1,
         hps.train.segment_size // hps.data.hop_length,
         **hps.model)
-    net_g.load_state_dict(torch.load(a.ckpt))
+    net_g.load_state_dict(load_checkpoint(a.ckpt, device))
     _ = net_g.eval()
 
     text2w2v = Text2W2V(hps.data.filter_length // 2 + 1,
     hps.train.segment_size // hps.data.hop_length,
     **hps_t2w2v.model)
-    text2w2v.load_state_dict(torch.load(a.ckpt_text2w2v))
+    text2w2v.load_state_dict(load_checkpoint(a.ckpt_text2w2v, device))
     text2w2v.eval()
   
     speechsr = SpeechSR48(h_sr48.data.n_mel_channels,