  params (Dict or FrozenDict) —
Dictionary containing the model parameters/weights.   prng_seed (jax.Array) —
Array containing random number generator key.   num_inference_steps (int, optional, defaults to 50) —
The number of denoising steps. More denoising steps usually lead to a higher quality image at the
expense of slower inference.   guidance_scale (float, optional, defaults to 7.5) —
A higher guidance scale value encourages the model to g