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

quick fix for not unwrapping model before getting config

parent b29141f8

audiolm_pytorch/trainer.py

+1 −1
Original line number Diff line number Diff line
@@ -257,7 +257,7 @@ class SoundStreamTrainer(nn.Module): 
        pkg = dict(

            model = self.accelerator.get_state_dict(self.soundstream),

            optim = self.optim.state_dict(),

            config = self.soundstream._configs,

            config = self.unwrapped_soundstream._configs,

            discr_optim = self.discr_optim.state_dict(),

            version = __version__

        )

audiolm_pytorch/version.py

+1 −1
Original line number Diff line number Diff line 
__version__ = '0.23.2'
 
__version__ = '0.23.3'