AutoencoderKLMagvit

The 3D variational autoencoder (VAE) model with KL loss used in EasyAnimate was introduced by Alibaba PAI.

The model can be loaded with the following code snippet.

from diffusers import AutoencoderKLMagvit

vae = AutoencoderKLMagvit.from_pretrained("alibaba-pai/EasyAnimateV5.1-12b-zh", subfolder="vae", torch_dtype=torch.float16).to("cuda")

AutoencoderKLMagvit

[[autodoc]] AutoencoderKLMagvit - decode - encode - all

AutoencoderKLOutput

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

DecoderOutput

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