+
    TV-jA  ã                   ó¨   € ^ 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)Úllama)ÚBaseModelArgsc                   ó2   a € ] tR t^t o R tV 3R ltRtV tR# )Ú	ModelArgsc                óv   € R V P                   R&   V P                   P                  R^ 4      V P                   R&   R# )FÚtie_word_embeddingsÚnum_attention_headsN)Útext_configÚget©Úselfs   &Úf/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_lm/models/pixtral.pyÚ__post_init__ÚModelArgs.__post_init__   s:   € Ø27ˆ×ÑÐ.Ñ/Ø26×2BÑ2B×2FÑ2FØ! 2ó3
ˆ×ÑÐ.Ó/ó    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   ø€ ÷ Wñ W™Yñ W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   r/   ÚModel.__init__   sD   ø€ Ü‰ÑÔØŒ	ØŸ/™/ˆŒÜ#Ÿkšk¬%¯/©/×*CÑ*CÀD×DTÑDTÓ*UÓVˆÖ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      # ))Úcacher6   )r1   )r   r5   r:   r6   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4      4      # )Úvision_towerNÚmulti_modal_projector)r   ÚlistÚitemsÚpopr   r   )r   Úweightss   &&r   ÚsanitizeÚModel.sanitize+   sD   € Ü ¤ g§m¡m£oÓ!6Ó7ˆØ�‰�N DÔ)Ø�‰Ð+¨TÔ2Ü”L Ó)Ó*Ð*r   c                óB   € V P                   P                  P                  # r-   )r1   ÚmodelÚlayersr   s   &r   rH   ÚModel.layers1   s   € à×"Ñ"×(Ñ(×/Ñ/Ð/r   )r*   r1   r   )NN)r    r!   r"   r#   r/   r;   rD   ÚpropertyrH   r%   r&   Ú__classcell__)r2   r   s   @@r   r(   r(      s9   ù‡ € ÷Wó W÷
ò 
ò+ð ñ0ó ÷0ð 0r   r(   )Údataclassesr   Útypingr   Úmlx.coreÚcorer7   Úmlx.nnÚnnÚ	mlx.utilsr   r   Ú r   Úbaser   r
   ÚModuler(   r   r   r   Ú<module>rV      sG   ðõ "Ý å Ý ß 2å Ý ð ô
�ó 
ó ð
ô0ˆB�I‰Iö 0r   