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

fix bug in complex stft discriminator output after switching to logits abs output

parent e3a4e897

audiolm_pytorch/soundstream.py

+4 −2
Original line number Diff line number Diff line
@@ -201,10 +201,12 @@ class ComplexSTFTDiscriminator(nn.Module): 


        complex_logits = self.final_conv(x)



        complex_logits_abs = torch.abs(complex_logits)



        if not return_intermediates:

            return complex_logits

            return complex_logits_abs



        return torch.abs(complex_logits), intermediates

        return complex_logits_abs, intermediates



# simulated complex stft discriminator

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

  version = '0.11.4',

  license='MIT',

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

  author = 'Phil Wang',