算力平台:
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