CogVideoXTransformer3DModel

CogVideoX 的 3D 数据扩散 Transformer 模型由清华大学和智谱AI在CogVideoX: 带有专家 Transformer 的文本到视频扩散模型中引入。

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

from diffusers import CogVideoXTransformer3DModel

vae = CogVideoXTransformer3DModel.from_pretrained("THUDM/CogVideoX-2b", subfolder="transformer", torch_dtype=torch.float16).to("cuda")

CogVideoXTransformer3DModel

[[autodoc]] CogVideoXTransformer3DModel

Transformer2DModelOutput

[[autodoc]] models.modeling_outputs.Transformer2DModelOutput