Hugging Face's logo

Diffusers documentation

ErnieImageTransformer2DModel

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v0.42.1).
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→