Update egs/visinger2/models.py

egs/visinger2/models.py

@@ -582,8 +582,9 @@ class Generator_Noise(torch.nn.Module):
         real = amp * torch.cos(phase)
         imag = amp * torch.sin(phase)
         spec = torch.cat([real, imag], 3)
-        istft_x = torch.istft(spec, self.fft_size, self.hop_size, self.win_size, self.window.to(amp), True, length=x.shape[2] * self.hop_size, return_complex=True)
-        istft_x = torch.view_as_real(istft_x)
+        istft_x = torch.view_as_complex(istft_x)
+
+        istft_x = torch.istft(spec, self.fft_size, self.hop_size, self.win_size, self.window.to(amp), True, length=x.shape[2] * self.hop_size, return_complex=False)
 
         return istft_x.unsqueeze(1)