updated the fusion method to use cat(Mel Spectrogram + Conditional Em…

…bedding, F0 helper)
IAHispano · Jan 9, 2025 · 2fa9ea8 · 2fa9ea8
1 parent 5ed5d41
commit 2fa9ea8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rvc/lib/algorithm/generators/refinegan.py b/rvc/lib/algorithm/generators/refinegan.py
@@ -458,8 +458,8 @@ def forward(self, mel: torch.Tensor, f0: torch.Tensor, g: torch.Tensor = None):
 
         if g is not None:
             # adding expanded speaker embedding
-            x += self.cond(g)
-        x = torch.cat([x, mel], dim=1)
+            mel += self.cond(g)
+        x = torch.cat([mel, x], dim=1)
 
         for ups, res, down in zip(
             self.upsample_blocks,