mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-29 07:22:12 +03:00
1.2 KiB
1.2 KiB
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