fix(rmvpe): pass device when loading torch model (svc-develop-team#301)

Author: quicksandznzn

modules/F0Predictor/rmvpe/inference.py

@@ -16,7 +16,7 @@ def __init__(self, model_path, device=None, dtype = torch.float32, hop_length=16
         else:
             self.device = device
         model = E2E0(4, 1, (2, 2))
-        ckpt = torch.load(model_path)
+        ckpt = torch.load(model_path, map_location=torch.device(self.device))
         model.load_state_dict(ckpt['model'])
         model = model.to(dtype).to(self.device)
         model.eval()
@@ -54,4 +54,4 @@ def infer_from_audio(self, audio, sample_rate=16000, thred=0.05, use_viterbi=Fal
         mel = mel_extractor(audio_res, center=True).to(self.dtype)
         hidden = self.mel2hidden(mel)
         f0 = self.decode(hidden.squeeze(0), thred=thred, use_viterbi=use_viterbi)
-        return f0
+        return f0