+
    œ°³j  ã                   óÊ   € ^ RI Ht ^RIHt ^RIHt ^RIHt ^RIH	t	 ^RI
HtHt ]! RR7      ] ! R	 R
]4      4       4       t]! RR7      ] ! R R]4      4       4       tRR
.tR# )é    )Ústrict)ÚPreTrainedConfig)ÚRopeParameters)Úauto_docstring)Úinterval)ÚCONFIG_MAPPINGÚ
AutoConfigzrhymes-ai/Aria)Ú
checkpointc                   ó  a a€ ] tR t^t oRtRtR.tRRRRRRRR	R
RRRRR	/tRR.R.3RRR.R.3RR.R.3/tRt	Rt
Rt^ t^ tRtRtRt]! RRR7      ! RR7      tRtRt^t^t^t^tR tRtR tRtR tRtR!t^t ^t!^t"V 3R" lt#R# t$V3R$ lt%R%t&Vt'V ;t(# )&ÚAriaTextConfiga%  
moe_num_experts (`int`, *optional*, defaults to 8):
    The number of experts in the MoE layer.
moe_topk (`int`, *optional*, defaults to 2):
    The number of top experts to route to for each token.
moe_num_shared_experts (`int`, *optional*, defaults to 2):
    The number of shared experts.
Ú	aria_textÚpast_key_valueszlayers.*.self_attn.q_projÚcolwisezlayers.*.self_attn.k_projzlayers.*.self_attn.v_projzlayers.*.self_attn.o_projÚrowwisez%layers.*.mlp.shared_experts.gate_projz#layers.*.mlp.shared_experts.up_projz%layers.*.mlp.shared_experts.down_projÚembed_tokensÚ	input_idsÚinputs_embedsÚlayersÚhidden_statesÚattention_maskÚnormi }  i   NÚsilui   g        g      ð?)ÚminÚmaxç{®Gáz”?)Údefaultg�íµ ÷Æ°>TFÚtext_configc                óÆ   <€ V P                   f#   V P                  V P                  ,          V n         V P                  f   V P                  V n        \        SV `  ! R/ VB  R # )N© )Úhead_dimÚhidden_sizeÚnum_attention_headsÚnum_key_value_headsÚsuperÚ__post_init__)ÚselfÚkwargsÚ	__class__s   &,€Úl/home/luca/screen-checking/.venv/lib/python3.14/site-packages/transformers/models/aria/configuration_aria.pyr%   ÚAriaTextConfig.__post_init__U   sO   ø€ Ø�=‰=Ò Ø ×,Ñ,°×0HÑ0HÕHˆDŒMØ×#Ñ#Ò+Ø'+×'?Ñ'?ˆDÔ$ä‰ÒÑ' Ô'ó    c                ó–   € V P                   V P                  ,          ^ 8w  d'   \        RV P                    RV P                   R24      hR# )zOPart of `@strict`-powered validation. Validates the architecture of the config.zThe hidden size (z6) is not a multiple of the number of attention heads (z).N)r!   r"   Ú
ValueError)r&   s   &r)   Úvalidate_architectureÚ$AriaTextConfig.validate_architecture]   sS   € à×Ñ˜d×6Ñ6Õ6¸!Ô;ÜØ# D×$4Ñ$4Ð#5ð 6Ø×2Ñ2Ð3°2ð7óð ñ <r+   c                óê  <€ V ^8„  d   Qh/ S[ ;R&   S[ ;R&   S[ ;R&   S[ ;R&   S[ ;R&   S[ R,          ;R&   S[;R&   S[ ;R	&   S[;R
&   S[;R&   S[;R&   S[ R,          ;R&   S[ R,          ;R&   S[ S[S[ ,          ,          R,          ;R&   S[ R,          ;R&   S[;R&   S[S[,          R,          ;R&   S[;R&   S[ S[,          R,          ;R&   S[;R&   S[ R,          ;R&   S[ ;R&   S[ ;R&   S[ ;R&   # )é   Ú
vocab_sizer!   Úintermediate_sizeÚnum_hidden_layersr"   Nr#   Ú
hidden_actÚmax_position_embeddingsÚinitializer_rangeÚrms_norm_epsÚ	use_cacheÚpad_token_idÚbos_token_idÚeos_token_idÚpretraining_tpÚtie_word_embeddingsÚrope_parametersÚattention_biasÚattention_dropoutÚmlp_biasr    Úmoe_num_expertsÚmoe_topkÚmoe_num_shared_experts)ÚintÚstrÚfloatÚboolÚlistr   Údict)ÚformatÚ__classdict__s   "€r)   Ú__annotate__ÚAriaTextConfig.__annotate__   s�  ø‡ ‚ ñ: Ññ; ñ< Ññ= ñ@ Ñ!ñA ñB ÑñC ñD Ñ!ñE ñF ˜t�Ñ*ñG ñH ÑñI ñJ !Ñ'ñK ñL ÑGñM ñN ÑñO ñP ÑñQ ñR ˜•*Ñ ñS ñT ˜•*Ñ ñU ñV ™™S�	•/ DÕ(Ñ,ñW ñX ˜$•JÑ"ñY ñZ Ñ%ñ[ ñ\ $¡dÕ*¨TÕ1Ñ8ñ] ñ^ Ñ ñ_ ñ` ™U•{ TÕ)Ñ/ña ñb Ññc ñd �D�jÑñe ñh Ññi ñj Ññk ñl  Ñ#òm r+   )r    r#   ))Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__doc__Ú
model_typeÚkeys_to_ignore_at_inferenceÚbase_model_tp_planÚbase_model_pp_planr2   r!   r3   r4   r"   r#   r5   r6   r   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   r    Úbase_config_keyrC   rD   rE   r%   r.   Ú__annotate_func__Ú__static_attributes__Ú__classdictcell__Ú__classcell__©r(   rM   s   @@r)   r   r      s'  ù‡ € ñð €JØ#4Ð"5Ðà# YØ# YØ# YØ# YØ/°Ø-¨yØ/°ðÐð 	˜+˜¨Ð(9Ð:Ø�OÐ%5Ð6¸Ð8IÐJØ�/Ð" _Ð$5Ð6ðÐð €JØ€Kà!ÐØÐØ!ÐØ&*ÐØ€JØ#'ÐÙ'¨C°SÕ9À$ÔGÐØ€LØ€IØ €LØ €LØ+,€LØ!"€NØ %ÐØ48€OØ €NØ,/ÐØ€HØ€HØ#€OØ€OØ€HØ"#Ðõ(ò÷A … r+   r   c                   óv   a a€ ] tR t^ft oRtRtRR/tR]R]/t	Rt
RtRtRt^	tRtR	tV 3R
 ltV3R ltRtVtV ;t# )Ú
AriaConfigze
projector_patch_to_query_dict (`dict`, *optional*):
    Mapping of patch sizes to query dimensions.
ÚariaÚimage_token_idÚimage_token_indexr   Úvision_configNr   Fc                ó$  <€ V P                   f   R^€RR/V n         V P                   P                  4        UUu/ uF  w  r#\        V4      \        V4      bK  	  uppV n         \        V P                   P	                  4       4      V n        \        V P                  \        4      '       dD   RV P                  R&   \        V P                  R,          ,          ! R/ V P                  B V n        M%V P                  f   \        R,          ! 4       V n        \        V P                  \        4      '       d-   RV P                  9   d   \        R/ V P                  B V n
        MV P                  f   \        4       V n
        \        SV `4  ! R/ VB  R # u uppi )NiÉ  i$  é   Úidefics3_visionrU   r   )Úprojector_patch_to_query_dictÚitemsrF   r   ÚvaluesÚ'max_value_projector_patch_to_query_dictÚ
isinstancerd   rK   r   r   r   r$   r%   )r&   r'   ÚkÚvr(   s   &,  €r)   r%   ÚAriaConfig.__post_init__|   sE  ø€ ð ×-Ñ-Ò5à�cØ�cð2ˆDÔ.ð JN×IkÑIk×IqÑIqÔIsÔ-tÑIsÁÀ¬c°!«f´c¸!³fªnÑIsÒ-tˆÔ*Ü7:¸4×;]Ñ;]×;dÑ;dÓ;fÓ7gˆÔ4ä�d×(Ñ(¬$×/Ò/Ø/@ˆD×Ñ˜|Ñ,Ü!/°×0BÑ0BÀ<Õ0PÖ!QÑ!gÐTX×TfÑTfÑ!gˆDÕØ×ÑÒ'Ü!/Ð0AÖ!BÓ!DˆDÔä�d×&Ñ&¬×-Ò-°,À$×BRÑBRÔ2RÜ-ÑA°×0@Ñ0@ÑAˆDÕØ×ÑÒ%Ü-Ó/ˆDÔä‰ÒÑ' Ô'ùó .us   ¸ Fc                óØ   <€ V ^8„  d   Qh/ S[ S[,          R,          ;R&   S[ S[,          R,          ;R&   S[S[S[,          ,          ;R&   S[ R,          ;R&   S[;R&   S[;R&   S[;R&   # )	r1   Nrd   r   Úvision_feature_layerrh   rc   r7   r>   )rK   r   r   rF   rJ   rH   rI   )rL   rM   s   "€r)   rN   ÚAriaConfig.__annotate__f   s�   ø‡ ‚ ñ Ñ*Õ*¨TÕ1Ñ8ñ ñ ™Õ&¨Õ-Ñ4ñ ñ  ¡¡S¥	�/Ñ.ñ! ñ" $(¨$¥;Ñ5ñ# ñ$ Ññ% ñ& Ñ#ñ' ñ( Ñ%ò) r+   )rk   rh   r   rd   éÿÿÿÿ)rP   rQ   rR   rS   rT   rU   Úattribute_mapr   r	   Úsub_configsrd   r   rq   rh   rc   r7   r>   r%   rZ   r[   r\   r]   r^   s   @@r)   r`   r`   f   sb   ù‡ € ñð
 €JàÐ-ð€Mð ! .°/À:ÐN€Kà48€MØ04€KØ,.ÐØ15Ð!ØÐØ#ÐØ %Ðõ(÷- … r+   r`   N)Úhuggingface_hub.dataclassesr   Úconfiguration_utilsr   Úmodeling_rope_utilsr   Úutilsr   Úutils.type_validatorsr   Úautor   r	   r   r`   Ú__all__r   r+   r)   Ú<module>r}      s   ðõ( /å 3Ý 1Ý #Ý -ß -ñ Ð+Ô,ØôDÐ%ó Dó ó -ðDñN Ð+Ô,Øô*(Ð!ó *(ó ó -ð*(ðZ Ð)Ð
*‚r+   