.float32'>

)


Parameters 

vae (FlaxAutoencoderKL) —
Variational Auto-Encoder (VAE) Model to encode and decode images to and from latent representations.


text_encoder (FlaxCLIPTextModel) —
Frozen text-encoder. Stable Diffusion uses the text portion of
CLIP,
specifically the clip-vit-large-patch14 variant.


tokenizer (CLIPTokenizer) —
Tokenizer of class
CLIPTokenizer.


unet (FlaxUNet2DConditionModel) — Conditional U-Net architecture to den