+
    UV-jÄ  ã                   ó®   € ^ RI Ht ^ RIHtHt ^RIHt ] ! R R]4      4       t] ! R R]4      4       t] ! R R	]4      4       t	] ! R
 R]4      4       t
R# )é    )Ú	dataclass)ÚListÚOptional)ÚBaseModelConfigc                   óP   a € ] tR t^t o RtRt^ t^ t^t^t	^àt
^ tRtV 3R ltRtV tR# )ÚVisionConfigé   é 8  g�íµ ÷Æ°>c                ó’   <€ V ^8„  d   Qh/ S[ ;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R	&   S[;R
&   # )é   Ú
model_typeÚhidden_sizeÚintermediate_sizeÚnum_hidden_layersÚnum_attention_headsÚnum_key_value_headsÚnum_channelsÚ
image_sizeÚ
patch_sizeÚlayer_norm_eps)ÚstrÚintÚfloat)ÚformatÚ__classdict__s   "€Úo/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_vlm/models/idefics2/config.pyÚ__annotate__ÚVisionConfig.__annotate__   s„   ø‡ ‚ á�Oñ ñ Ññ ñ Ñ"ñ	 ñ
 Ññ ñ Ñ!ñ ñ Ñ ñ ñ Ññ ñ Ññ ñ Ññ ñ Ñ ò ó    © N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r   r   r   r   r   r   r   r   r   Ú__annotate_func__Ú__static_attributes__Ú__classdictcell__©r   s   @r   r   r      s?   ø‡ € ð €KØ"ÐØÐØ!ÐØ ÐØ€LØ€JØ€JØ €N÷ ƒ r   r   c                   ó^   a € ] tR t^t o RtRt^ t^ t^tRt	Rt
RtRtRtRtR tV 3R	 ltR
tV tR# )Ú
TextConfigr	   r
   gñhãˆµøä>i}  g    €„.AFi €  c                óH   € V P                   f   V P                  V n         R # R # ©N)r   r   ©Úselfs   &r   Ú__post_init__ÚTextConfig.__post_init__$   s!   € Ø×#Ñ#Ò+Ø'+×'?Ñ'?ˆDÖ$ñ ,r   c                óª   <€ V ^8„  d   Qh/ S[ ;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R	&   S[;R
&   S[;R&   S[;R&   # )r   r   r   r   r   r   r   Úrms_norm_epsÚ
vocab_sizeÚ
rope_thetaÚrope_traditionalÚmax_position_embeddingsÚtie_word_embeddings)r   r   r   Úbool)r   r   s   "€r   r   ÚTextConfig.__annotate__   sž   ø‡ ‚ á�Oñ ñ Ññ ñ Ñ"ñ	 ñ
 Ññ ñ Ñ!ñ ñ Ñ ñ ñ Ññ ñ Ññ ñ Ñ!ñ ñ Ñ"ñ ñ !Ñ(ñ ñ Ñ%ò r   )r   N)r!   r"   r#   r$   r   r   r   r   r   r2   r3   r4   r5   r6   r7   r/   r%   r&   r'   r(   s   @r   r*   r*      sR   ø‡ € ð €KØ"ÐØÐØ!ÐØ ÐØ€LØ€JØ!€JØ"ÐØ#(ÐØ %Ðò@÷ ƒ r   r*   c                   ó@   a € ] tR t^)t o ^t^t^`t^t^@tV 3R lt	Rt
V tR# )ÚPerceiverConfigc                ób   <€ V ^8„  d   Qh/ S[ ;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   # )r   r   r   Úresampler_depthÚresampler_head_dimÚresampler_n_headsÚresampler_n_latents)r   r   )r   r   s   "€r   r   ÚPerceiverConfig.__annotate__)   sP   ø‡ ‚ á�Oñ ñ Ñ ñ ñ Ññ	 ñ
 Ñ ñ ñ Ññ ñ Ñ!ò r   r    N)r!   r"   r#   r$   r   r=   r>   r?   r@   r%   r&   r'   r(   s   @r   r;   r;   )   s+   ø‡ € ð  !ÐØ€OØ ÐØÐØ!Ð÷ ƒ r   r;   c                   óF   a € ] tR t^3t o RtRtRtRtRtR t	V 3R lt
RtV tR# )ÚModelConfigi}  i€Q Nc                óH   € V P                   f   V P                  V n         R # R # r,   )Úimage_token_indexÚimage_token_idr-   s   &r   r/   ÚModelConfig.__post_init__?   s!   € Ø×!Ñ!Ò)Ø%)×%8Ñ%8ˆDÖ"ñ *r   c                ó¶   <€ V ^8„  d   Qh/ S[ ;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[S[,          ;R&   S[S[S[,          ,          ;R	&   # )
r   Útext_configÚvision_configÚperceiver_configr   Úignore_indexrF   r3   rE   Úeos_token_id)r*   r   r;   r   r   r   r   )r   r   s   "€r   r   ÚModelConfig.__annotate__3   s„   ø‡ ‚ áÑñ ñ  Ññ ñ &Ñ%ñ	 ñ
 �Oñ ñ Ññ ñ Ññ ñ Ññ ñ  ¡•}Ñ+ñ ñ ™4¡�9Õ%Ñ,ò r   )rE   iœÿÿÿ)r!   r"   r#   r$   rL   rF   r3   rE   rM   r/   r%   r&   r'   r(   s   @r   rC   rC   3   s-   ø‡ € ð €LØ€NØ€JØ'+ÐØ(,€Lò9÷ ƒ r   rC   N)Údataclassesr   Útypingr   r   Úbaser   r   r*   r;   rC   r    r   r   Ú<module>rR      s}   ðÝ !ß !å "ð ô
!�?ó 
!ó ð
!ð ô@�ó @ó ð@ð& ô"�oó "ó ð"ð ô9�/ó 9ó ò9r   