QwenTest / pythonProject /diffusers-main /docs /source /en /api /models /autoencoder_kl_wan.md
xiaoanyu123's picture
Add files using upload-large-folder tool
e2bcd96 verified
|
Raw History Blame Contribute Delete
1.12 kB

AutoencoderKLWan

The 3D variational autoencoder (VAE) model with KL loss used in Wan 2.1 by the Alibaba Wan Team.

The model can be loaded with the following code snippet.

from diffusers import AutoencoderKLWan

vae = AutoencoderKLWan.from_pretrained("Wan-AI/Wan2.1-T2V-1.3B-Diffusers", subfolder="vae", torch_dtype=torch.float32)

AutoencoderKLWan

[[autodoc]] AutoencoderKLWan

  • decode
  • all

DecoderOutput

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