visinger2-nomidi

Sleeping

kevinwang676 commited on Jun 25, 2024

Commit

b6087a4

verified ·

1 Parent(s): 274cf03

Update egs/visinger2/models.py

Files changed (1) hide show

egs/visinger2/models.py CHANGED Viewed

@@ -582,7 +582,6 @@ 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.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)

         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=False)