feat: return direct and wet signals separately

modules/fx.py

@@ -907,10 +907,9 @@ class SendFXsAndSum(FX):
             di = x
 
         if len(self.params) == 0:
-            return reduce(
+            return di, reduce(
                 lambda x, y: x[..., : y.shape[-1]] + y[..., : x.shape[-1]],
                 map(lambda f: f(x), self.effects),
-                di,
             )
 
         def f(states, ps):
@@ -927,15 +926,21 @@ class SendFXsAndSum(FX):
                 ),
             )
 
-        return reduce(
-            f,
-            zip(
-                self.effects,
-                [self.params[f"sends_{i}"] for i in range(len(self.effects) - 1)]
-                + [None],
-            ),
-            (di, x.unsqueeze(0).expand(len(self.effects), -1, -1, -1)),
-        )[0]
+        return (
+            di,
+            reduce(
+                f,
+                zip(
+                    self.effects,
+                    [self.params[f"sends_{i}"] for i in range(len(self.effects) - 1)]
+                    + [None],
+                ),
+                (
+                    torch.zeros_like(x),
+                    x.unsqueeze(0).expand(len(self.effects), -1, -1, -1),
+                ),
+            )[0],
+        )
 
 
 class UniLossLess(nn.Module):