Method encode

tokenizer/vqgan/model.py:69–73

(self, x)

            print(f"Unexpected Keys in State Dict: {unexpected}")

    def encode(self, x):
        h = self.encoder(x)
        h = self.quant_conv(h)
        quant, emb_loss, info = self.quantize(h)
        return quant, emb_loss, info

    def decode(self, quant):
        quant = self.post_quant_conv(quant)