mirror of
https://github.com/huggingface/diffusers.git
synced 2025-12-16 01:14:47 +08:00
1.1 KiB
1.1 KiB
CogView4Transformer2DModel
A Diffusion Transformer model for 2D data from CogView4
The model can be loaded with the following code snippet.
from diffusers import CogView4Transformer2DModel
transformer = CogView4Transformer2DModel.from_pretrained("THUDM/CogView4-6B", subfolder="transformer", torch_dtype=torch.bfloat16).to("cuda")
CogView4Transformer2DModel
autodoc CogView4Transformer2DModel
Transformer2DModelOutput
autodoc models.modeling_outputs.Transformer2DModelOutput