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

use base class from audiolm-pytorch for type validation

parent 0503f7d1

musiclm_pytorch/musiclm_pytorch.py

+2 −1
Original line number Diff line number Diff line
@@ -5,6 +5,7 @@ from torch import nn, einsum 
from torchaudio.transforms import Spectrogram, TimeStretch, FrequencyMasking, TimeMasking



from audiolm_pytorch import AudioLM

from audiolm_pytorch.utils import AudioConditionerBase



from x_clip.tokenizer import tokenizer

from vector_quantize_pytorch import ResidualVQ
@@ -448,7 +449,7 @@ class MuLaN(nn.Module): 
# music lm



@beartype

class MuLaNEmbedQuantizer(nn.Module):

class MuLaNEmbedQuantizer(AudioConditionerBase):

    def __init__(

        self,

        mulan: MuLaN,

setup.py

+2 −2
Original line number Diff line number Diff line
@@ -3,7 +3,7 @@ from setuptools import setup, find_packages 
setup(

  name = 'musiclm-pytorch',

  packages = find_packages(exclude=[]),

  version = '0.0.5',

  version = '0.0.6',

  license='MIT',

  description = 'MusicLM - AudioLM + Audio CLIP to text to music synthesis',

  author = 'Phil Wang',
@@ -19,7 +19,7 @@ setup( 
    'contrastive learning'

  ],

  install_requires=[

    'audiolm-pytorch>=0.8.1',

    'audiolm-pytorch>=0.9.0',

    'beartype',

    'einops>=0.4',

    'vector-quantize-pytorch>=1.0.0',