Commit 6123c770 authored by Phil Wang's avatar Phil Wang
Browse files

make a guess to suppress a warning, but need expert guidance here

parent 8afeb585

audiolm_pytorch/soundstream.py

+3 −3
Original line number Diff line number Diff line
@@ -450,7 +450,7 @@ class SoundStream(nn.Module): 
        enc_cycle_dilations = (1, 3, 9),

        dec_cycle_dilations = (1, 3, 9),

        multi_spectral_window_powers_of_two = tuple(range(6, 12)),

        num_mel_bins = 64,

        multi_spectral_n_mels = (8, 16, 32, 64, 64, 64, 64),

        recon_loss_weight = 1.,

        multi_spectral_recon_loss_weight = 1.,

        adversarial_loss_weight = 1.,
@@ -547,7 +547,7 @@ class SoundStream(nn.Module): 
        self.mel_spec_transforms = nn.ModuleList([])

        self.mel_spec_recon_alphas = []



        for powers in multi_spectral_window_powers_of_two:

        for powers, n_mels in zip(multi_spectral_window_powers_of_two, multi_spectral_n_mels):

            win_length = 2 ** powers

            alpha = (win_length / 2) ** 0.5
@@ -556,7 +556,7 @@ class SoundStream(nn.Module): 
                n_fft = win_length,

                win_length = win_length,

                hop_length = win_length // 4,

                n_mels = num_mel_bins

                n_mels = n_mels,

            )



            self.mel_spec_transforms.append(melspec_transform)

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

  version = '0.11.8',

  license='MIT',

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

  author = 'Phil Wang',