Model tokenizer: CLIPTokenizer unet: FlaxUNet2DConditionModel scheduler: Union safety_checker: FlaxStableDiffusionSafetyChecker feature_extractor: CLIPImageProcessor dtype: dtype = <class 'jax.numpy.float32'>  )   Parameters   vae (FlaxAutoencoderKL) —
Variational Auto-Encoder (VAE) model to encode and decode images to and from latent representations.   text_encoder (FlaxCLIPTextModel) —
Frozen text-encoder (clip-vit-large-patch14).   tokenizer