custom_diffusers/docs/source/en/api/models/autoencoderkl_mochi.md

AutoencoderKLMochi

The 3D variational autoencoder (VAE) model with KL loss used in Mochi was introduced in Mochi 1 Preview by Tsinghua University & ZhipuAI.

The model can be loaded with the following code snippet.

from diffusers import AutoencoderKLMochi

vae = AutoencoderKLMochi.from_pretrained("genmo/mochi-1-preview", subfolder="vae", torch_dtype=torch.float32).to("cuda")

AutoencoderKLMochi

[[autodoc]] AutoencoderKLMochi - decode - all

DecoderOutput

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