jnp.ndarray) —
Array representing an image batch to be used as the starting point.   params (Dict or FrozenDict) —
Dictionary containing the model parameters/weights.   prng_seed (jax.Array or jax.Array) —
Array containing random number generator key.   strength (float, optional, defaults to 0.8) —
Indicates extent to transform the reference image. Must be between 0 and 1. image is used as a
starting point and more noise is added the higher the