+
    TV-j  ã                   ó¨   € ^ RI Ht ^ RIHt ^ RIHt ^ RIHt ^ RI	H
t
Ht ^RIHt ^RIHt ] ! R R]4      4       t ! R	 R
]P"                  4      tR# )é    )Ú	dataclass)ÚOptionalN)Útree_flattenÚtree_unflatten)Úgemma3_text)ÚBaseModelArgsc                   ó6   a € ] tR t^t o RtR tV 3R ltRtV tR# )Ú	ModelArgsi@  c                óÜ   € V P                   V P                  R &   V P                  P                  R^4      V P                  R&   V P                  P                  R^4      V P                  R&   R# )Ú
vocab_sizeÚnum_attention_headsÚnum_key_value_headsN)r   Útext_configÚget©Úselfs   &Úe/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_lm/models/gemma3.pyÚ__post_init__ÚModelArgs.__post_init__   sf   € Ø)-¯©ˆ×Ñ˜Ñ&Ø26×2BÑ2B×2FÑ2FØ! 1ó3
ˆ×ÑÐ.Ñ/ð 37×2BÑ2B×2FÑ2FØ! 1ó3
ˆ×ÑÐ.Ó/ó    c                ó>   <€ V ^8„  d   Qh/ S[ ;R&   S[;R&   S[;R&   # )é   Ú
model_typer   r   )ÚstrÚdictÚint)ÚformatÚ__classdict__s   "€r   Ú__annotate__ÚModelArgs.__annotate__   s)   ø‡ ‚ á�Oñ ñ Ññ ñ Ñò	 r   © N)	Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r   r   Ú__annotate_func__Ú__static_attributes__Ú__classdictcell__)r   s   @r   r
   r
      s   ø‡ € ð €Jò
÷ ƒ r   r
   c                   óp   a a€ ] tR t^t oV3R lV 3R lltR	V3R lR lltR t]R 4       tR t	Rt
VtV ;t# )
ÚModelc                ó    <€ V ^8„  d   QhRS[ /# )r   Úargs)r
   )r   r   s   "€r   r   ÚModel.__annotate__   s   ø€ ÷ 
ñ 
™Yñ 
r   c                óÖ   <€ \         SV `  4        Wn        VP                  V n        \        P
                  ! \        P                  P                  VP                  4      4      V n	        R # ©N)
ÚsuperÚ__init__r,   r   r   r*   r
   Ú	from_dictr   Úlanguage_model)r   r,   Ú	__class__s   &&€r   r1   ÚModel.__init__   sK   ø€ Ü‰ÑÔØŒ	ØŸ/™/ˆŒÜ)×/Ò/Ü×!Ñ!×+Ñ+¨D×,<Ñ,<Ó=ó
ˆÖr   c                ó^   <€ V ^8„  d   QhRS[ P                  RS[S[ P                  ,          /# )r   ÚinputsÚinput_embeddings)ÚmxÚarrayr   )r   r   s   "€r   r   r-   '   s,   ø€ ÷ 
ñ 
á—‘ð
ñ #¡2§8¡8Õ,ñ	
r   c                ó(   € V P                  WVR 7      # ))Úcacher8   )r3   )r   r7   r<   r8   s   &&&&r   Ú__call__ÚModel.__call__'   s"   € ð ×"Ñ"ØÐ2Bð #ó 
ð 	
r   c                ól  € \        \        VP                  4       4      4      pVP                  R R4       VP                  RR4       \	        \        VR,          4      4      pV P                  P                  V4      p\        \        VP                  4       4      4      VR&   \	        \        V4      4      # )Úvision_towerNÚmulti_modal_projectorr3   )r   ÚlistÚitemsÚpopr   r   r3   Úsanitize)r   ÚweightsÚ
lm_weightss   && r   rE   ÚModel.sanitize1   s�   € Ü ¤ g§m¡m£oÓ!6Ó7ˆØ�‰�N DÔ)Ø�‰Ð+¨TÔ2Üœ, wÐ/?Õ'@ÓAÓBˆ
Ø×(Ñ(×1Ñ1°*Ó=ˆ
Ü$2´4¸
×8HÑ8HÓ8JÓ3KÓ$LˆÐ Ñ!Ü”L Ó)Ó*Ð*r   c                ó.   € V P                   P                  # r/   )r3   Úlayersr   s   &r   rJ   ÚModel.layers:   s   € à×"Ñ"×)Ñ)Ð)r   c                ó6   € V P                   P                  4       # r/   )r3   Ú
make_cacher   s   &r   rM   ÚModel.make_cache>   s   € Ø×"Ñ"×-Ñ-Ó/Ð/r   )r,   r3   r   )NN)r"   r#   r$   r%   r1   r=   rE   ÚpropertyrJ   rM   r'   r(   Ú__classcell__)r4   r   s   @@r   r*   r*      s<   ù‡ € ÷
ó 
÷
ò 
ò+ð ñ*ó ð*÷0ò 0r   r*   )Údataclassesr   Útypingr   Úmlx.coreÚcorer9   Úmlx.nnÚnnÚ	mlx.utilsr   r   Ú r   Úbaser   r
   ÚModuler*   r!   r   r   Ú<module>r[      sG   ðõ "Ý å Ý ß 2å Ý ð ô
�ó 
ó ð
ô!0ˆB�I‰Iö !0r   