s 

prompt_ids (jnp.array) —
The prompt or prompts to guide the image generation.


image (jnp.array) —
Array representing the ControlNet input condition. ControlNet use this input condition to generate
guidance to Unet.


params (Dict or FrozenDict) — Dictionary containing the model parameters/weights


prng_seed (jax.random.KeyArray or jax.Array) — Array containing random number generator key


num_inference_steps (int, optional, defaults to 