bugfinx: mps not support float64

This commit is contained in:
w-okada 2023-07-19 22:08:56 +09:00
parent 43a1583731
commit a6b1f093e1

View File

@ -143,7 +143,7 @@ class Pipeline(object):
f0_up_key,
silence_front=silence_front,
)
pitch = torch.tensor(pitch[-n_frames:], device=self.device).unsqueeze(0).long()
pitch = torch.tensor(pitch[-n_frames:], device=self.device, dtype=torch.float).unsqueeze(0).long()
except IndexError as e: # NOQA
raise NotEnoughDataExtimateF0()