Update infer_tools/infer_tool.py

infer_tools/infer_tool.py

@@ -179,11 +179,11 @@ class Svc:
 
 def getitem(item):
     max_frames = hparams['max_frames']
-    spec = torch.Tensor(item['mel'])[:max_frames]
-    mel2ph = torch.LongTensor(item['mel2ph'])[:max_frames] if 'mel2ph' in item else None
+    spec = torch.Tensor(item['mel']).cpu()[:max_frames]
+    mel2ph = torch.LongTensor(item['mel2ph']).cpu()[:max_frames] if 'mel2ph' in item else None
     f0, uv = norm_interp_f0(item["f0"][:max_frames], hparams)
-    hubert = torch.Tensor(item['hubert'][:hparams['max_input_tokens']])
-    pitch = torch.LongTensor(item.get("pitch"))[:max_frames]
+    hubert = torch.Tensor(item['hubert'][:hparams['max_input_tokens']]).cpu()
+    pitch = torch.LongTensor(item.get("pitch")).cpu()[:max_frames]
     sample = {
         "id": item['id'],
         "spk_id": item['spk_id'],