Method decode_code

tokenizer/vqgan/model.py:80–83

(self, code_b, shape, channel_first=True)

        return dec

    def decode_code(self, code_b, shape, channel_first=True):
        quant_b = self.quantize.get_codebook_entry(code_b, shape, channel_first)
        dec = self.decode(quant_b)
        return dec

    def forward(self, input):
        quant, diff, _ = self.encode(input)