Commit 455def25 authored by Phil Wang's avatar Phil Wang
Browse files

fix when turning off local attention for sound stream

parent d5890a24

audiolm_pytorch/soundstream.py

+2 −2
Original line number Diff line number Diff line
@@ -384,8 +384,8 @@ class SoundStream(nn.Module): 
        return [

            *self.encoder.parameters(),

            *self.decoder.parameters(),

            *self.encoder_attn.parameters(),

            *self.decoder_attn.parameters()

            *(self.encoder_attn.parameters() if exists(self.encoder_attn) else []),

            *(self.decoder_attn.parameters() if exists(self.decoder_attn) else [])

        ]



    @property

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

  version = '0.6.3',

  license='MIT',

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

  author = 'Phil Wang',