Unverified Commit 554ffa16 authored by Eren Gölge's avatar Eren Gölge Committed by GitHub
Browse files

Fix stft

parent d0151be9

audiolm_pytorch/soundstream.py

+1 −1
Original line number Diff line number Diff line
@@ -152,7 +152,7 @@ class STFTDiscriminator(nn.Module): 
        H = 256 samples

        '''



        x = torch.view_as_complex(torch.stft(x,1024, hop_length=256,win_length=1024))

        x = torch.view_as_complex(torch.stft(x,1024, hop_length=256,win_length=1024, return_complex=False))

        x = rearrange(x, 'b ... -> b 1 ...')



        intermediates = []