Skip to content

AutoencoderKLMochi

Mochi中使用的带有KL损失的3D变分自编码器(VAE)模型由清华大学和智谱AI在Mochi 1 Preview中引入。

可以使用以下代码片段加载该模型。

python
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