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

also prepare encodec wrapper for use in natural speech 2 repo

parent 99bf7cb6

audiolm_pytorch/encodec.py

+3 −0
Original line number Diff line number Diff line
@@ -81,6 +81,9 @@ class EncodecWrapper(nn.Module): 
        #   back to b n anyways, but we'll keep this as a temporary hack just to make things work for now

        return rearrange(result, 'b n -> b 1 n')



    def decode(self, emb):

        return self.model.decoder(emb)



    def _decode_frame(self, quantized_indices):

        # The following code is hacked in from self.model._decode_frame() (Encodec version 0.1.1) where we assume we've

        # already unwrapped the EncodedFrame

audiolm_pytorch/version.py

+1 −1
Original line number Diff line number Diff line 
__version__ = '0.27.0'
 
__version__ = '0.27.1'