Diffusers documentation

AutoencoderKLHunyuanVideo15

Diffusers

Начало работы

Diffusers Установка Быстрый старт Интерфейс командной строки

Вывод

Использование конвейеров диффузии

Адаптеры и персонализация

Оптимизация и масштабирование

Базовая производительность

Скорость и память

Распределённый вывод

Квантование

Ускорение для конкретного оборудования

Методы сообщества

Модульные Diffusers

Обучение и дообучение

Ресурсы

Внести вклад

API

Основные классы

Модульный

Загрузчики

Модели

Конвейеры

Планировщики

Внутренние классы

Join the Hugging Face community

and get access to the augmented documentation experience

Collaborate on models, datasets and Spaces

Faster examples with accelerated inference

Switch between documentation themes

to get started

AutoencoderKLHunyuanVideo15

Модель трёхмерного вариационного автоэнкодера (VAE) с потерей KL, используемая в HunyuanVideo1.5 от Tencent.

Модель можно загрузить с помощью следующего фрагмента кода.

from diffusers import AutoencoderKLHunyuanVideo15

vae = AutoencoderKLHunyuanVideo15.from_pretrained("hunyuanvideo-community/HunyuanVideo-1.5-Diffusers-480p_t2v", subfolder="vae", dtype=torch.float32)

# make sure to enable tiling to avoid OOM
vae.enable_tiling()

AutoencoderKLHunyuanVideo15

класс diffusers.AutoencoderKLHunyuanVideo15

< source >

( in_channels: int = 3out_channels: int = 3latent_channels: int = 32block_out_channels: tuple[int] = (128, 256, 512, 1024, 1024)layers_per_block: int = 2spatial_compression_ratio: int = 16temporal_compression_ratio: int = 4downsample_match_channel: bool = Trueupsample_match_channel: bool = Truescaling_factor: float = 1.03682 )

Модель VAE с потерей KL для кодирования видео в латенты и декодирования латентных представлений в видео. Используется для HunyuanVideo-1.5.

Эта модель наследуется от ModelMixin. См. документацию суперкласса для его универсальных методов, реализованных для всех моделей (например, загрузка или сохранение).

decode

< source >

( z: Tensorreturn_dict: bool = True ) → ~models.vae.DecoderOutput or tuple

Параметры

z (torch.Tensor) — Входная партия латентных векторов.
return_dict (bool, необязательный, по умолчанию True) — Определяет, следует ли возвращать ~models.vae.DecoderOutput вместо обычного кортежа.

Возвращает

~models.vae.DecoderOutput или tuple

Если return_dict равно True, возвращается ~models.vae.DecoderOutput, иначе возвращается обычный tuple.

Декодировать пакет изображений.

encode

< source >

( x: Tensorreturn_dict: bool = True )

Параметры

x (torch.Tensor) — Входная партия изображений.
return_dict (bool, необязательный, по умолчанию True) — Определяет, следует ли возвращать ~models.autoencoder_kl.AutoencoderKLOutput вместо обычного кортежа.

Кодирует пакет изображений в латенты.

enable_tiling

< source >

Параметры

tile_sample_min_height (int, необязательный) — Минимальная высота, необходимая для того, чтобы образец был разделён на плитки по размерности высоты.
tile_sample_min_width (int, необязательный) — Минимальная ширина, необходимая для того, чтобы образец был разделён на плитки по размерности ширины.
tile_latent_min_height (int, необязательный) — Минимальная высота, необходимая для того, чтобы латентное представление разделялось на тайлы по размерности высоты.
tile_latent_min_width (int, необязательный) — Минимальная ширина, необходимая для того, чтобы латентное представление разделялось на плитки по ширине.

Включает тайловое декодирование VAE. При включении этого параметра VAE разбивает входной тензор на тайлы, чтобы выполнять декодирование и кодирование в несколько шагов. Это полезно для экономии значительного количества памяти и для обработки изображений большего размера.

forward

< source >

( sample: Tensorsample_posterior: bool = Falsereturn_dict: bool = Truegenerator: typing.Optional[torch.Generator] = None ) → ~models.vae.DecoderOutput or tuple

Параметры

sample (torch.Tensor) — Входной образец.
sample_posterior (bool, необязательный, по умолчанию False) — Следует ли извлекать выборку из апостериорного распределения.
return_dict (bool, необязательный, по умолчанию True) — Определяет, следует ли возвращать DecoderOutput вместо обычного кортежа.
generator (torch.Generator, необязательный) — torch.Generator для обеспечения детерминированности выборки.