eThPSrSSKrSSKJr SSKJrJrJrJr SSK r SSK r SSK J r SSK J r SSKJrJrJrJr SS KJr SS KJrJrJr SS KJrJrJr S S KJrJrJ r \RB"\"5r#\"SS\55r$\"SS\55r%\"SS\55r&S\ RNS\ RN4Sjr(S\ RNS\ RN4Sjr)S\ S\*4Sjr+SNS\\*\4S\,4Sjjr-"SS \ R\5r/"S!S"\ R`5r1"S#S$\ R\5r2"S%S&\ R\5r3"S'S(\ R\5r4"S)S*\ R\5r5"S+S,\ R\5r6"S-S.\ R\5r7"S/S0\ R\5r8"S1S2\ R\5r9"S3S4\ R\5r:S5\90r;"S6S7\ R\5r<"S8S9\ R\5r="S:S;\ R\5r>"S<S=\ R\5r?"S>S?\ R\5r@"S@SA\ R\5rA\"SBSC\55rB\"SDSE9"SFSG\B55rC\"SHSE9"SISJ\B55rD\"SKSL\B55rE/SMQrFg)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) AlignConfigAlignTextConfigAlignVisionConfigc\rSrSr%SrSr\\R\ S'Sr \\R\ S'Sr \\ \R\ S'Sr g)AlignVisionModelOutput)a 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____firstlineno____doc__rrtorch FloatTensor__annotations__rrr__static_attributes__r `/var/www/auris/envauris/lib/python3.13/site-packages/transformers/models/align/modeling_align.pyrr)sN 15L(5,,-459x 1 1298