Commit 2725ae89 authored by Phil Wang's avatar Phil Wang
Browse files

fix a bug thanks to @eonglints

parent 5f964966

audiolm_pytorch/audiolm_pytorch.py

+1 −1
Original line number Diff line number Diff line
@@ -109,7 +109,7 @@ class MultiScaleDiscriminator(nn.Module): 
            chan_out = min(curr_channels * 4, chan_max)



            self.conv_layers.append(nn.Sequential(

                nn.Conv1d(curr_channels, chan_out, 8, stride = 4, padding = 4),

                nn.Conv1d(curr_channels, chan_out, 8, stride = 4, padding = 4, groups = groups),

                leaky_relu()

            ))

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

  version = '0.0.20',

  license='MIT',

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

  author = 'Phil Wang',