Method encode

ldm/models/autoencoder.py:289–292

(self, x)

        self.embed_dim = embed_dim

    def encode(self, x):# VQModel的quantize写在encoder里,VQModelInterface则将其写在decoder里
        h = self.encoder(x)
        h = self.quant_conv(h)
        return h

    def decode(self, h, force_not_quantize=False):
        # also go through quantization layer