hidream_image_transformer.md

HiDreamImageTransformer2DModel

A Transformer model for image-like data from HiDream-I1.

The model can be loaded with the following code snippet.

fromdiffusersimportHiDreamImageTransformer2DModeltransformer=HiDreamImageTransformer2DModel.from_pretrained("HiDream-ai/HiDream-I1-Full", subfolder="transformer", torch_dtype=torch.bfloat16)

HiDreamImageTransformer2DModel

[[autodoc]] HiDreamImageTransformer2DModel

Transformer2DModelOutput

[[autodoc]] models.modeling_outputs.Transformer2DModelOutput