/
ai-forever
/
diffusers-new
Обзор
Документация
Войти
/
ai-forever
/
diffusers-new
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
main
src/diffusers/models/modeling_outputs.py
17 строк
512 B
Suraj Patil
Add SVD (#5895)
29 ноя 2023, 21:13
Не верифицирован
29 ноя 2023, 21:13
63f767e
Код
Авторство
О чём код?
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