Commit 627d6fee authored by Phil Wang's avatar Phil Wang
Browse files

allow for ability to turn off multi spectral recon loss

parent b6ff57d9

audiolm_pytorch/soundstream.py

+5 −4
Original line number Diff line number Diff line
@@ -699,6 +699,7 @@ class SoundStream(nn.Module): 


        multi_spectral_recon_loss = 0



        if self.multi_spectral_recon_loss_weight > 0:

            for mel_transform, alpha in zip(self.mel_spec_transforms, self.mel_spec_recon_alphas):

                orig_mel, recon_mel = map(mel_transform, (orig_x, recon_x))

                log_orig_mel, log_recon_mel = map(log, (orig_mel, recon_mel))

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

  version = '0.11.6',

  license='MIT',

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

  author = 'Phil Wang',