r Quantized Variational Auto-Encoder (VAE) Model to encode and decode images to and from latent
representations.


text_encoder (CLIPTextModel) —
Frozen text-encoder. VQ Diffusion uses the text portion of
CLIP, specifically
the clip-vit-base-patch32 variant.


tokenizer (CLIPTokenizer) —
Tokenizer of class
CLIPTokenizer.


transformer (Transformer2DModel) —
Conditional transformer to denoise the encoded image latents.


scheduler (VQDiffusionSc