diffusers / docs /source /en /api /models /autoencoder_kl_hunyuanimage.md
AbdulElahGwaith's picture
Upload folder using huggingface_hub
ac2243f verified

AutoencoderKLHunyuanImage

The 2D variational autoencoder (VAE) model with KL loss used in [HunyuanImage2.1].

The model can be loaded with the following code snippet.

from diffusers import AutoencoderKLHunyuanImage

vae = AutoencoderKLHunyuanImage.from_pretrained("hunyuanvideo-community/HunyuanImage-2.1-Diffusers", subfolder="vae", torch_dtype=torch.bfloat16)

AutoencoderKLHunyuanImage

[[autodoc]] AutoencoderKLHunyuanImage

  • decode
  • all

DecoderOutput

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