ixin) —
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 that estimates whether generated images could be considered offensive or harmful.
Please, refer to the model card for details.


feature_extractor (CLIPImageProcess