AutoencoderKLAllegro

在Allegro中使用的带有KL损失的3D变分自编码器（VAE）模型，由RhymesAI在Allegro: Open the Black Box of Commercial-Level Video Generation Model中引入。

可以使用以下代码片段加载该模型。

from diffusers import AutoencoderKLAllegro

vae = AutoencoderKLCogVideoX.from_pretrained("rhymes-ai/Allegro", subfolder="vae", torch_dtype=torch.float32).to("cuda")

AutoencoderKLAllegro

[[autodoc]] AutoencoderKLAllegro - decode - encode - all

AutoencoderKLOutput

[[autodoc]] models.autoencoders.autoencoder_kl.AutoencoderKLOutput

DecoderOutput

[[autodoc]] models.autoencoders.vae.DecoderOutput