# ErnieImageTransformer2DModel

Um modelo Transformer para dados semelhantes a imagens do [ERNIE-Image](https://huggingface.co/baidu/ERNIE-Image).

Um modelo Transformer para dados semelhantes a imagens do [ERNIE-Image-Turbo](https://huggingface.co/baidu/ERNIE-Image-Turbo).

## ErnieImageTransformer2DModel[[diffusers.ErnieImageTransformer2DModel]]

#### diffusers.ErnieImageTransformer2DModel[[diffusers.ErnieImageTransformer2DModel]]

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

[Fonte](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/models/transformers/transformer_ernie_image.py#L296)

#### forward[[diffusers.ErnieImageTransformer2DModel.forward]]

```python
forward(hidden_states: Tensor, timestep: Tensor, text_bth: Tensor, text_lens: Tensor, return_dict: bool = True)
```

[Fonte](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/models/transformers/transformer_ernie_image.py#L348)

**Parâmetros:**

hidden_states (`torch.Tensor` de forma `(batch_size, in_channels, height, width)`) : `hidden_states` de entrada.

timestep (`torch.LongTensor`) : Usado para indicar o passo de desruidamento.

text_bth (`torch.Tensor`) : embeddings de texto condicionais (embeddings calculados a partir das condições de entrada, como prompts) a serem usados, com forma `(batch_size, text_length, embed_dims)`.

text_lens (`torch.Tensor`) : Comprimentos das sequências de texto por amostra usados para construir a máscara de atenção.

return_dict (`bool`, *opcional*, padrão `True`) : Indica se deve retornar um `~models.transformer_2d.Transformer2DModelOutput` em vez de uma tupla simples.

**Retorna:**

Se `return_dict` for True, um `ErnieImageTransformer2DModelOutput` é retornado; caso contrário, uma `tuple` em que o primeiro elemento é o tensor da amostra.

O método forward do [ErnieImageTransformer2DModel](/docs/diffusers/v0.43.0/pt/api/models/ernie_image_transformer2d#diffusers.ErnieImageTransformer2DModel).