mirror of
https://github.com/huggingface/diffusers.git
synced 2025-12-06 12:34:13 +08:00
1.1 KiB
1.1 KiB
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