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