diff --git a/server/voice_changer/SoVitsSvc40/SoVitsSvc40.py b/server/voice_changer/SoVitsSvc40/SoVitsSvc40.py index 61390060..c90e2f73 100644 --- a/server/voice_changer/SoVitsSvc40/SoVitsSvc40.py +++ b/server/voice_changer/SoVitsSvc40/SoVitsSvc40.py @@ -204,7 +204,7 @@ class SoVitsSvc40: print("not only one speaker found.", speaker) else: cluster_c = cluster.get_cluster_center_result(self.cluster_model, c.cpu().numpy().T, speaker[0]).T - cluster_c = torch.FloatTensor(cluster_c).cpu() + cluster_c = torch.FloatTensor(cluster_c).to(dev) c = self.settings.clusterInferRatio * cluster_c + (1 - self.settings.clusterInferRatio) * c c = c.unsqueeze(0)