Update egs/visinger2/models.py

egs/visinger2/models.py

@@ -578,7 +578,7 @@ class Generator_Noise(torch.nn.Module):
         istft_x = self.net(istft_x) * mask
     
         # Compute amplitude and random phase
-        amp = self.istft_amplitude(istft_x).unsqueeze(-1)
+        amp = self.istft_amplitude(istft_x)
         phase = (torch.rand(amp.shape) * 2 * 3.14 - 3.14).to(amp.device)
     
         # Calculate real and imaginary parts
@@ -595,6 +595,7 @@ class Generator_Noise(torch.nn.Module):
 
 
 
+
 class LayerNorm(nn.Module):
     def __init__(self, channels, eps=1e-5):
         super().__init__()