Method encode

diffusers/src/diffusers/models/autoencoders/vq_model.py:130–137

(self, x: torch.Tensor, return_dict: bool = True)

    @apply_forward_hook
    def encode(self, x: torch.Tensor, return_dict: bool = True) -> VQEncoderOutput:
        h = self.encoder(x)
        h = self.quant_conv(h)

        if not return_dict:
            return (h,)

        return VQEncoderOutput(latents=h)

    @apply_forward_hook
    def decode(