diffusers/models/modeling_outputs.py

from dataclasses import dataclass

from ..utils import BaseOutput


@dataclass
class AutoencoderKLOutput(BaseOutput):
    """

    Output of AutoencoderKL encoding method.



    Args:

        latent_dist (`DiagonalGaussianDistribution`):

            Encoded outputs of `Encoder` represented as the mean and logvar of `DiagonalGaussianDistribution`.

            `DiagonalGaussianDistribution` allows for sampling latents from the distribution.

    """

    latent_dist: "DiagonalGaussianDistribution"  # noqa: F821