Commit fe97568c authored by Phil Wang's avatar Phil Wang
Browse files

default back to stft discriminator using torch.complex64, due to... 

default back to stft discriminator using torch.complex64, due to https://github.com/lucidrains/audiolm-pytorch/issues/64
parent c6c8299a

audiolm_pytorch/soundstream.py

+1 −1
Original line number Diff line number Diff line
@@ -462,7 +462,7 @@ class SoundStream(nn.Module): 
        attn_window_size = 128,

        attn_dim_head = 64,

        attn_heads = 8,

        use_complex_stft_discriminator = False

        use_complex_stft_discriminator = True

    ):

        super().__init__()

        self.target_sample_hz = target_sample_hz # for resampling on the fly

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.7.5',

  version = '0.7.6',

  license='MIT',

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

  author = 'Phil Wang',