Commit 4daad9e7 authored by Phil Wang's avatar Phil Wang
Browse files

expose the soundstream configs as a property

parent 8cb662b9

audiolm_pytorch/soundstream.py

+4 −0
Original line number Diff line number Diff line
@@ -510,6 +510,10 @@ class SoundStream(nn.Module): 
        self.adversarial_loss_weight = adversarial_loss_weight

        self.feature_loss_weight = feature_loss_weight



    @property

    def configs(self):

        return pickle.loads(self._configs)



    def decode_from_codebook_indices(self, quantized_indices):

        codes = self.rq.get_codes_from_indices(quantized_indices)

        x = reduce(codes, 'q ... -> ...', 'sum')

audiolm_pytorch/version.py

+1 −1
Original line number Diff line number Diff line 
__version__ = '0.17.0'
 
__version__ = '0.17.1'