support timbre confusion

utils/util.py

@@ -29,6 +29,20 @@ from utils.hparam import HParams
 import logging
 from logging import handlers
 
+def interpolate_embeddings(embedding1, embedding2, alpha):
+        """
+        Interpolate between two embeddings.
+
+        :param embedding1: First embedding vector.
+        :param embedding2: Second embedding vector.
+        :param alpha: Interpolation factor (0 to 1).
+        :return: Interpolated embedding.
+        """
+        if not 0 <= alpha <= 1:
+            raise ValueError("Alpha should be between 0 and 1")
+        fused_embedding = (1 - alpha) * np.array(embedding1.cpu()) + alpha * np.array(embedding2.cpu())
+        device = torch.device("cuda" if torch.cuda.is_available() else "cpu") 
+        return torch.from_numpy(fused_embedding).to(device)
 
 def str2bool(v):
     """Used in argparse.ArgumentParser.add_argument to indicate