Commit 6bf92afa authored by Phil Wang's avatar Phil Wang
Browse files

add more informative error messsage

parent 62ad1710

audiolm_pytorch/audiolm_pytorch.py

+1 −1
Original line number Diff line number Diff line
@@ -826,7 +826,7 @@ class SemanticTransformerWrapper(nn.Module): 
        super().__init__()

        self.wav2vec = wav2vec

        self.transformer = transformer

        assert self.wav2vec.codebook_size == transformer.num_semantic_tokens

        assert self.wav2vec.codebook_size == transformer.num_semantic_tokens, f'num_semantic_tokens on SemanticTransformer must be set to {self.wav2vec.codebook_size}'



        self.unique_consecutive = unique_consecutive

        self.pad_id = pad_id

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

  version = '0.1.2',

  license='MIT',

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

  author = 'Phil Wang',