MochiTransformer3DModel

Genmo 在 Mochi-1 Preview 中介绍了一种用于 3D 类视频数据的扩散 Transformer 模型。

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

from diffusers import MochiTransformer3DModel

vae = MochiTransformer3DModel.from_pretrained("genmo/mochi-1-preview", subfolder="transformer", torch_dtype=torch.float16).to("cuda")

MochiTransformer3DModel

[[autodoc]] MochiTransformer3DModel

Transformer2DModelOutput

[[autodoc]] models.modeling_outputs.Transformer2DModelOutput