+
    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Ht ^RIHt ] ! R R]4      4       t ! R	 R
]P$                  4      tR# )é    )Ú	dataclass)ÚOptionalN)Útree_flattenÚtree_unflatten)ÚllamaÚ
ministral3)ÚBaseModelArgsc                   ó2   a € ] tR t^t o R tV 3R ltRtV tR# )Ú	ModelArgsc                óJ   € R V P                   9  d   RV P                   R &   R# R# )Útie_word_embeddingsFN)Útext_config©Úselfs   &Úg/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_lm/models/mistral3.pyÚ__post_init__ÚModelArgs.__post_init__   s&   € Ø ¨×(8Ñ(8Ô8Ø6;ˆD×ÑÐ2Ó3ñ 9ó    c                ó2   <€ V ^8„  d   Qh/ S[ ;R&   S[;R&   # )é   Ú
model_typer   )ÚstrÚdict)ÚformatÚ__classdict__s   "€r   Ú__annotate__ÚModelArgs.__annotate__   s   ø‡ ‚ á�Oñ ñ Ñò r   © N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r   Ú__annotate_func__Ú__static_attributes__Ú__classdictcell__)r   s   @r   r   r      s   ø‡ € ò
<÷ ƒ r   r   c                   ój   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	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        VP                  P                  R 4      R8X  dE   \        P                  ! \        P                  P                  VP                  4      4      V n
        R# \        P                  ! \        P                  P                  VP                  4      4      V n
        R# )r   r   N)ÚsuperÚ__init__r)   r   r   Úgetr   r'   r   Ú	from_dictÚlanguage_modelr   )r   r)   Ú	__class__s   &&€r   r-   ÚModel.__init__   s�   ø€ Ü‰ÑÔØŒ	ØŸ/™/ˆŒØ×Ñ×Ñ Ó-°Ô=Ü",×"2Ò"2Ü×$Ñ$×.Ñ.¨t×/?Ñ/?Ó@ó#ˆDÖô #(§+¢+Ü—‘×)Ñ)¨$×*:Ñ*:Ó;ó#ˆ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      # ))Úcacher5   )r0   )r   r4   r9   r5   s   &&&&r   Ú__call__ÚModel.__call__&   s"   € ð ×"Ñ"ØÐ2Bð #ó 
ð 	
r   c                ó(  € \        \        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      VR&   \	        \        V4      4      # )Úvision_towerNÚmulti_modal_projectorr0   )r   ÚlistÚitemsÚpopr   r   r0   Úsanitize)r   ÚweightsÚ
lm_weightss   && r   rB   ÚModel.sanitize0   su   € Ü ¤ g§m¡m£oÓ!6Ó7ˆØ�‰�N DÔ)Ø�‰Ð+¨TÔ2Üœ, wÐ/?Õ'@ÓAÓBˆ
Ø$(×$7Ñ$7×$@Ñ$@ÀÓ$LˆÐ Ñ!Ü”L Ó)Ó*Ð*r   c                óB   € V P                   P                  P                  # )N)r0   ÚmodelÚlayersr   s   &r   rH   ÚModel.layers8   s   € à×"Ñ"×(Ñ(×/Ñ/Ð/r   )r)   r0   r   )NN)r   r    r!   r"   r-   r:   rB   ÚpropertyrH   r$   r%   Ú__classcell__)r1   r   s   @@r   r'   r'      s7   ù‡ € ÷ó ÷
ò 
ò+ð ñ0ó ÷0ð 0r   r'   )Údataclassesr   Útypingr   Úmlx.coreÚcorer6   Úmlx.nnÚnnÚ	mlx.utilsr   r   Ú r   r   Úbaser	   r   ÚModuler'   r   r   r   Ú<module>rV      sG   ðõ "Ý å Ý ß 2ç Ý ð ô<�ó <ó ð<ô"0ˆB�I‰Iö "0r   