unet to denoise the encoded image latents. Can be one of
FlaxDDIMScheduler, FlaxLMSDiscreteScheduler, FlaxPNDMScheduler, or
FlaxDPMSolverMultistepScheduler.   safety_checker (FlaxStableDiffusionSafetyChecker) —
Classification module that estimates whether generated images could be considered offensive or harmful.
Please refer to the model card for more details
about a model’s potential harms.   feature_extractor (CLIPImageProcessor) —
A CLIPIma