Unverified Commit b0e10403 authored by Phil Wang's avatar Phil Wang Committed by GitHub
Browse files

Merge pull request #30 from erogol/stft_fix 

Add required STFT argument
parents d0151be9 554ffa16

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 = []