dule and refer to the Flax documentation for all matter related to
general usage and behavior.
Finally, this model supports inherent JAX features such as:
Just-In-Time (JIT) compilation
Automatic Differentiation
Vectorization
Parallelization

FlaxDecoderOutput
	

class diffusers.models.vae_flax.FlaxDecoderOutput

<
source
>
(
sample: ndarray

)


Parameters 

sample (jnp.ndarray of shape (batch_size, num_channels, height, width)) —
Decoded outp