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

fix save command not using the one on the trainer class, for saving optimizer state

parent 0247f2cc

musiclm_pytorch/trainer.py

+1 −2
Original line number Diff line number Diff line
@@ -333,9 +333,8 @@ class MuLaNTrainer(nn.Module): 
        # save model every so often



        if self.is_main and not (steps % self.save_model_every):

            state_dict = self.mulan.state_dict()

            model_path = str(self.results_folder / f'mulan.{steps}.pt')

            torch.save(state_dict, model_path)

            self.save(model_path)



            self.print(f'{steps}: saving model to {str(self.results_folder)}')

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

  version = '0.0.25',

  license='MIT',

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

  author = 'Phil Wang',