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

fix a small bug thanks to @Lunariz

parent 6ed3aad4

audiolm_pytorch/soundstream.py

+1 −0
Original line number Diff line number Diff line
@@ -145,6 +145,7 @@ class ComplexConv2d(nn.Module): 


    def forward(self, x):

        weight, bias = map(torch.view_as_complex, (self.weight, self.bias))

        print(x.dtype, weight.dtype, x.device, weight.device)

        return F.conv2d(x, weight, bias, stride = self.stride, padding = self.padding)



def ComplexSTFTResidualUnit(chan_in, chan_out, strides):

audiolm_pytorch/trainer.py

+1 −2
Original line number Diff line number Diff line
@@ -951,8 +951,7 @@ class FineTransformerTrainer(nn.Module): 
                folder,

                max_length = data_max_length,

                target_sample_hz = soundstream.target_sample_hz,

                seq_len_multiple_of = soundstream.seq_len_multiple_of,

                normalize = dataset_normalize

                seq_len_multiple_of = soundstream.seq_len_multiple_of

            )



        self.ds_fields = None

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

  version = '0.11.16',

  license='MIT',

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

  author = 'Phil Wang',