er (CLIPTokenizer) —
Tokenizer of class
CLIPTokenizer.


unet (FlaxUNet2DConditionModel) — Conditional U-Net architecture to denoise the encoded image latents.


controlnet (FlaxControlNetModel —
Provides additional conditioning to the unet during the denoising process.


scheduler (SchedulerMixin) —
A scheduler to be used in combination with unet to denoise the encoded image latents. Can be one of
FlaxDDIMScheduler, FlaxLMSDiscreteScheduler, F