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

fix embarrassing error, thanks to @alexdemartos

parent d244a6c9

audiolm_pytorch/soundstream.py

+1 −1
Original line number Diff line number Diff line
@@ -491,7 +491,7 @@ class SoundStream(nn.Module): 
        codes = self.rq.get_codes_from_indices(quantized_indices)

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



        x = self.decoder_attn(x) + x

        x = self.decoder_attn(x)

        x = rearrange(x, 'b n c -> b c n')

        return self.decoder(x)