Commit 920a8a46 authored by Phil Wang's avatar Phil Wang
Browse files

fix stft discriminator

parent c5ea4618

audiolm_pytorch/soundstream.py

+1 −1
Original line number Diff line number Diff line
@@ -164,7 +164,7 @@ class STFTDiscriminator(nn.Module): 


        x = self.stft(x)

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

        x = torch.cat((x.real, x.imag), dim = 1).detach()

        x = torch.cat((x.real, x.imag), dim = 1)



        intermediates = []

setup.py

+1 −1
Original line number Diff line number Diff line
@@ -3,7 +3,7 @@ from setuptools import setup, find_packages 
setup(

  name = 'audiolm-pytorch',

  packages = find_packages(exclude=[]),

  version = '0.3.1',

  version = '0.3.2',

  license='MIT',

  description = 'AudioLM - Language Modeling Approach to Audio Generation from Google Research - Pytorch',

  author = 'Phil Wang',