ay mid_block_res_sample: Array  )   Parameters   down_block_res_samples (jnp.ndarray) —   mid_block_res_sample (jnp.ndarray) —    The output of FlaxControlNetModel.  replace  < source > ( **updates  )    “Returns a new object replacing the specified fields with new values.
