Hugging Face's logo

Diffusers documentation

ErnieImageTransformer2DModel

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

ErnieImageTransformer2DModel

ERNIE-Image'dan görüntü benzeri veriler için bir Transformer modeli.

ERNIE-Image-Turbo kaynaklı görüntü benzeri veriler için bir Transformer modeli.

ErnieImageTransformer2DModel

sınıf diffusers.ErnieImageTransformer2DModel

< >

( hidden_size: int = 3072num_attention_heads: int = 24num_layers: int = 24ffn_hidden_size: int = 8192in_channels: int = 128out_channels: int = 128patch_size: int = 1text_in_dim: int = 2560rope_theta: int = 256rope_axes_dim: typing.Tuple[int, int, int] = (32, 48, 48)eps: float = 1e-06qk_layernorm: bool = True )

forward

< >

( hidden_states: Tensortimestep: Tensortext_bth: Tensortext_lens: Tensorreturn_dict: bool = True )

Parametreler

  • hidden_states (torch.Tensor, şekli (batch_size, in_channels, height, width)) — Girdi hidden_states.
  • timestep (torch.LongTensor) — Gürültü azaltma adımını belirtmek için kullanılır.
  • text_bth (torch.Tensor) — Kullanılacak koşullu metin gömme değerleri (prompt gibi girdi koşullarından hesaplanan gömme değerleri), şekli (batch_size, text_length, embed_dims).
  • text_lens (torch.Tensor) — Dikkat maskesini oluşturmak için kullanılan örnek başına metin dizisi uzunlukları.
  • return_dict (bool, isteğe bağlı, varsayılan True) — Düz bir tuple yerine ~models.transformer_2d.Transformer2DModelOutput döndürülüp döndürülmemesi.

ErnieImageTransformer2DModel ileri yöntemi.

Update on GitHub

←EasyAnimateTransformer3DModel Flux2Transformer2DModel→