bCi (SrSSKrSSKJr SSKJrJrJrJr SSK r SSK J r SSK J r SSK Jr SS KJrJrJrJr SS KJrJr SS KJrJrJr SS KJrJrJrJrJ r S SK!J"r"J#r#J$r$ \ RJ"\&5r'\\"SS9"SS\555r(\\"SS9"SS\555r)\\"SS\555r*S\ RVS\ RV4Sjr,S\ RVS\ RV4Sjr-S\$S\.4Sjr/SYS \\.\04S!\14S"jjr2"S#S$\ Rf5r4"S%S&\ Rj5r6"S'S(\ Rf5r7"S)S*\ Rf5r8"S+S,\ Rf5r9"S-S.\ Rf5r:"S/S0\ Rf5r;"S1S2\ Rf5r<"S3S4\ Rf5r=SZS5\ RfS6\ RVS7\ RVS8\ RVS9\\ RVS:\>S;\>S<\\ RV4S=jjr?"S>S?\ Rf5r@"S@SA\ Rf5rA"SBSC\ Rf5rB"SDSE\ Rf5rC"SFSG\ Rf5rD"SHSI\5rE"SJSK\ Rf5rF"SLSM\ Rf5rG\"SNSO\55rH\"SPS9"SQSR\H55rI\"SSS9"STSU\H55rJ\"SVSW\H55rK/SXQrLg)[zPyTorch ALIGN model.N) dataclass)AnyCallableOptionalUnion)nn)ACT2FN)GradientCheckpointingLayer)BaseModelOutputBaseModelOutputWithNoAttentionBaseModelOutputWithPooling(BaseModelOutputWithPoolingAndNoAttention)ALL_ATTENTION_FUNCTIONSPreTrainedModel)apply_chunking_to_forward find_pruneable_heads_and_indicesprune_linear_layer) ModelOutputauto_docstringcan_return_tuplefilter_out_non_signature_kwargslogging) AlignConfigAlignTextConfigAlignVisionConfigz} Base class for vision model's outputs that also contains image embeddings of the pooling of the last hidden states. ) custom_introc\rSrSr%SrSr\\R\ S'Sr \\R\ S'Sr \\ \R\ S'Sr g)AlignVisionModelOutput)z 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. N image_embedslast_hidden_state hidden_states)__name__ __module__ __qualname____firstlineno____doc__r"rtorch FloatTensor__annotations__r#r$tuple__static_attributes__r%b/home/james-whalen/.local/lib/python3.13/site-packages/transformers/models/align/modeling_align.pyr r )sN 15L(5,,-459x 1 1298