t_encoder: FlaxCLIPTextModel
tokenizer: CLIPTokenizer
unet: FlaxUNet2DConditionModel
scheduler: typing.Union[diffusers.schedulers.scheduling_ddim_flax.FlaxDDIMScheduler, diffusers.schedulers.scheduling_pndm_flax.FlaxPNDMScheduler, diffusers.schedulers.scheduling_lms_discrete_flax.FlaxLMSDiscreteScheduler, diffusers.schedulers.scheduling_dpmsolver_multistep_flax.FlaxDPMSolverMultistepScheduler]
safety_checker: FlaxStableDiffusionSafetyChecker
fe