Commit 47ee1c39 authored by Phil Wang's avatar Phil Wang
Browse files

potentially fix https://github.com/lucidrains/audiolm-pytorch/issues/64

parent edb8f779

audiolm_pytorch/soundstream.py

+2 −2
Original line number Diff line number Diff line
@@ -241,8 +241,8 @@ class ComplexModReLU(nn.Module): 
    def forward(self, x):

        real, imag = x.unbind(dim = 1)



        x_abs = (real ** 2 + imag ** 2).clamp(min = 1e-5).sqrt()

        x_angle = torch.atan(imag / real.clamp(min = 1e-5))

        x_abs = (real ** 2 + imag ** 2).clamp(min = 1e-8).sqrt()

        x_angle = torch.atan2(imag, real)



        new_real = F.relu(x_abs + self.b)

        new_imag = x_angle.exp()

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

  version = '0.7.7',

  license='MIT',

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

  author = 'Phil Wang',