mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-27 17:22:53 +03:00
1.3 KiB
1.3 KiB
CogView3PlusTransformer2DModel
A Diffusion Transformer model for 2D data from CogView3Plus was introduced in CogView3: Finer and Faster Text-to-Image Generation via Relay Diffusion by Tsinghua University & ZhipuAI.
The model can be loaded with the following code snippet.
from diffusers import CogView3PlusTransformer2DModel
transformer = CogView3PlusTransformer2DModel.from_pretrained("THUDM/CogView3Plus-3b", subfolder="transformer", torch_dtype=torch.bfloat16).to("cuda")
CogView3PlusTransformer2DModel
autodoc CogView3PlusTransformer2DModel
Transformer2DModelOutput
autodoc models.modeling_outputs.Transformer2DModelOutput