zer) —
A CLIPTokenizer to tokenize text.   unet (FlaxUNet2DConditionModel) —
A FlaxUNet2DConditionModel to denoise the encoded image latents.   scheduler (SchedulerMixin) —
A scheduler to be used in combination with unet to denoise the encoded image latents. Can be one of
FlaxDDIMScheduler, FlaxLMSDiscreteScheduler, FlaxPNDMScheduler, or
FlaxDPMSolverMultistepScheduler.   safety_checker (FlaxStableDiffusionSafetyChecker) —
Classification module