18 lines
512 B
Python
18 lines
512 B
Python
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
|