 

sample (jnp.ndarray of shape (batch_size, num_channels, height, width)) —
Hidden states conditioned on encoder_hidden_states input. Output of last layer of model.




replace

<
source
>
(
**updates

)



“Returns a new object replacing the specified fields with new values.

FlaxUNet2DConditionModel
	

class diffusers.FlaxUNet2DConditionModel

<
source
>
(
sample_size: int = 32
in_channels: int = 4
out_channels: int = 4
down_block_types: typ