Commit 0503f7d1 authored by Phil Wang's avatar Phil Wang
Browse files

override parameters, so if mulan embed quantizer is passed to audiolm... 

override parameters, so if mulan embed quantizer is passed to audiolm trainers, mulan is not trained
parent c40aee30

musiclm_pytorch/musiclm_pytorch.py

+3 −0
Original line number Diff line number Diff line
@@ -494,6 +494,9 @@ class MuLaNEmbedQuantizer(nn.Module): 


        self.set_default_namespace(namespaces[0])



    def parameters(self):

        return self.cond_embeddings.parameters()



    def set_default_namespace(self, namespace):

        self._default_namespace = namespace

setup.py

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

  version = '0.0.5',

  license='MIT',

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

  author = 'Phil Wang',