bugfix: cluster

This commit is contained in:
wataru 2023-03-20 11:13:19 +09:00
parent fd2a8d7736
commit ae132e0c0f

View File

@ -204,7 +204,7 @@ class SoVitsSvc40:
print("not only one speaker found.", speaker) print("not only one speaker found.", speaker)
else: else:
cluster_c = cluster.get_cluster_center_result(self.cluster_model, c.cpu().numpy().T, speaker[0]).T 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 = self.settings.clusterInferRatio * cluster_c + (1 - self.settings.clusterInferRatio) * c
c = c.unsqueeze(0) c = c.unsqueeze(0)