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

make sure one can detect easily whether audio conditioner is present for AudioLM

parent 15151252

audiolm_pytorch/audiolm_pytorch.py

+2 −0
Original line number Diff line number Diff line
@@ -1513,6 +1513,8 @@ class AudioLM(nn.Module): 
    ):

        super().__init__()



        self.audio_conditioner = audio_conditioner



        assert semantic_transformer.num_semantic_tokens == coarse_transformer.num_semantic_tokens

        assert coarse_transformer.codebook_size == fine_transformer.codebook_size

        assert coarse_transformer.num_coarse_quantizers == fine_transformer.num_coarse_quantizers

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

  version = '0.9.2',

  license='MIT',

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

  author = 'Phil Wang',