From ae132e0c0f522cedd1b64b8cbb168bafeb201af7 Mon Sep 17 00:00:00 2001 From: wataru Date: Mon, 20 Mar 2023 11:13:19 +0900 Subject: [PATCH] bugfix: cluster --- server/voice_changer/SoVitsSvc40/SoVitsSvc40.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)