Skip to content

CogView3PlusTransformer2DModel

一个用于2D数据的扩散Transformer模型,来自CogView3Plus,由清华大学和智谱AI在CogView3: 通过中继扩散实现更精细和更快的文本到图像生成中引入。

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

python
from diffusers import CogView3PlusTransformer2DModel

vae = CogView3PlusTransformer2DModel.from_pretrained("THUDM/CogView3Plus-3b", subfolder="transformer", torch_dtype=torch.bfloat16).to("cuda")

CogView3PlusTransformer2DModel

[[autodoc]] CogView3PlusTransformer2DModel

Transformer2DModelOutput

[[autodoc]] models.modeling_outputs.Transformer2DModelOutput