Commit 41861d49 authored by Phil Wang's avatar Phil Wang
Browse files

helpful error message

parent 2ebe3541

audiolm_pytorch/audiolm_pytorch.py

+3 −0
Original line number Diff line number Diff line
@@ -374,6 +374,8 @@ class Transformer(nn.Module): 
    ):

        super().__init__()

        assert not (cross_attend and cond_as_self_attn_prefix)

        self.dim_context = default(dim_context, dim)



        self.cond_as_self_attn_prefix = cond_as_self_attn_prefix



        self.grad_shrink = partial(grad_shrink, alpha = grad_shrink_alpha)
@@ -399,6 +401,7 @@ class Transformer(nn.Module): 
        context_mask = None

    ):

        assert not (self.cond_as_self_attn_prefix and not exists(context))

        assert not (exists(context) and context.shape[-1] != self.dim_context), f'you had specified a conditioning dimension of {self.dim_context}, yet what was received by the transformer has dimension of {context.shape[-1]}'



        n, device = x.shape[1], x.device

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

  version = '0.9.5',

  license='MIT',

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

  author = 'Phil Wang',