UhRdZddlZddlmZddlmZmZmZmZddl Z ddl Z ddl m Z ddl m Z ddlmZmZmZmZdd lmZdd lmZmZmZdd lmZmZmZd d lmZmZm Z ejBe"Z#eGddeZ$eGddeZ%eGddeZ&de jNde jNfdZ(de jNde jNfdZ)de de*fdZ+dNdee*efde,fdZ-Gdd e j\Z/Gd!d"e j`Z1Gd#d$e j\Z2Gd%d&e j\Z3Gd'd(e j\Z4Gd)d*e j\Z5Gd+d,e j\Z6Gd-d.e j\Z7Gd/d0e j\Z8Gd1d2e j\Z9Gd3d4e j\Z:d5e9iZ;Gd6d7e j\Z<Gd8d9e j\Z=Gd:d;e j\Z>Gd<d=e j\Z?Gd>d?e j\Z@Gd@dAe j\ZAeGdBdCeZBedDEGdFdGeBZCedHEGdIdJeBZDeGdKdLeBZEgdMZFy)OzPyTorch ALIGN model.N) dataclass)AnyOptionalTupleUnion)nn)ACT2FN)BaseModelOutputWithNoAttention)BaseModelOutputWithPastAndCrossAttentions,BaseModelOutputWithPoolingAndCrossAttentions(BaseModelOutputWithPoolingAndNoAttention)PreTrainedModel)apply_chunking_to_forward find_pruneable_heads_and_indicesprune_linear_layer) ModelOutputauto_docstringlogging) AlignConfigAlignTextConfigAlignVisionConfigceZdZUdZdZeejed<dZ eejed<dZ ee ejed<y)AlignVisionModelOutputa+ Base class for vision model's outputs that also contains image embeddings of the pooling of the last hidden states. Args: image_embeds (`torch.FloatTensor` of shape `(batch_size, output_dim)` *optional* returned when model is initialized with `with_projection=True`): The image embeddings obtained by applying the projection layer to the pooler_output. last_hidden_state (`torch.FloatTensor` of shape `(batch_size, sequence_length, hidden_size)`): Sequence of hidden-states at the output of the last layer of the model. hidden_states (`tuple(torch.FloatTensor)`, *optional*, returned when `output_hidden_states=True` is passed or when `config.output_hidden_states=True`): Tuple of `torch.FloatTensor` (one for the output of the embeddings, if the model has an embedding layer, + one for the output of each layer) of shape `(batch_size, sequence_length, hidden_size)`. Hidden-states of the model at the output of each layer plus the optional initial embedding outputs. N image_embedslast_hidden_state hidden_states) __name__ __module__ __qualname____doc__rrtorch FloatTensor__annotations__rrrz/var/www/catia.catastroantioquia-mas.com/valormas/lib/python3.12/site-packages/transformers/models/align/modeling_align.pyrr)sN 15L(5,,-459x 1 1298