eDiffusionPipelineOutput or tuple  Parameters   prompt_ids (jnp.ndarray) —
The prompt or prompts to guide the image generation.   image (jnp.ndarray) —
Array representing the ControlNet input condition to provide guidance to the unet for generation.   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, defa