ing_pndm_flax.FlaxPNDMScheduler, diffusers.schedulers.scheduling_lms_discrete_flax.FlaxLMSDiscreteScheduler, diffusers.schedulers.scheduling_dpmsolver_multistep_flax.FlaxDPMSolverMultistepScheduler]
safety_checker: FlaxStableDiffusionSafetyChecker
feature_extractor: CLIPFeatureExtractor
dtype: dtype = <class 'jax.numpy.float32'>

)


Parameters 

vae (FlaxAutoencoderKL) —
Variational Auto-Encoder (VAE) Model to encode and decode images to and f