+
    TV-j€	  ã                   óò   € ^ RI Ht ^ RIHtHt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IHt ^RIHt ] ! R	 R
]4      4       t ! R R]
P,                  4      t ! R R]
P,                  4      tR# )é    )Ú	dataclass)ÚAnyÚOptionalÚUnionN)Útree_flattenÚtree_unflatten)ÚBaseModelArgs)ÚDeepseekV3Model)ÚModel©Ú	ModelArgsc                   ó6   a € ] tR t^t o RtR tV 3R ltRtV tR# )r   Úkimi_k25c                ó”   € \        V P                  \        4      '       d(   \        P                  ! V P                  4      V n        R # R # ©N)Ú
isinstanceÚtext_configÚdictÚ
TextConfigÚ	from_dict©Úselfs   &Úg/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_lm/models/kimi_k25.pyÚ__post_init__ÚModelArgs.__post_init__   s3   € Ü�d×&Ñ&¬×-Ò-Ü)×3Ò3°D×4DÑ4DÓEˆDÖñ .ó    c                óH   <€ V ^8„  d   Qh/ S[ S[S[3,          ;R&   S[;R&   # )é   r   Ú
model_type)r   r   r   Ústr)ÚformatÚ__classdict__s   "€r   Ú__annotate__ÚModelArgs.__annotate__   s'   ø‡ ‚ á‘z¡4Ð'Õ(Ñ(ñ ñ Ñ ò r   )r   N)	Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r   r   Ú__annotate_func__Ú__static_attributes__Ú__classdictcell__)r"   s   @r   r   r      s   ø‡ € ð !€JòF÷ ƒ r   r   c                   óT   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VtV ;t# )ÚLanguageModelc                ó    <€ V ^8„  d   QhRS[ /# ©r   Úconfig)r   )r!   r"   s   "€r   r#   ÚLanguageModel.__annotate__   s   ø€ ÷ Tñ T™zñ Tr   c                ó´   <€ \         SV `  4        Wn        \        V4      V n        \
        P                  ! VP                  VP                  R R7      V n	        R# )F)ÚbiasN)
ÚsuperÚ__init__Úargsr
   ÚmodelÚnnÚLinearÚhidden_sizeÚ
vocab_sizeÚlm_head©r   r0   Ú	__class__s   &&€r   r5   ÚLanguageModel.__init__   s?   ø€ Ü‰ÑÔØŒ	Ü$ VÓ,ˆŒ
Ü—y’y ×!3Ñ!3°V×5FÑ5FÈUÔSˆŽr   c                óJ   <€ V ^8„  d   QhRS[ P                  RS[S[,          /# ©r   ÚinputsÚcache©ÚmxÚarrayr   r   )r!   r"   s   "€r   r#   r1   !   s'   ø€ ÷ !ñ !á—‘ð!ñ ™�}ñ!r   c                óF   € V P                  W4      pV P                  V4      # r   )r7   r<   )r   rB   rC   Úouts   &&& r   Ú__call__ÚLanguageModel.__call__!   s!   € ð
 �j‰j˜Ó'ˆØ�|‰|˜CÓ Ð r   )r6   r<   r7   r   )	r%   r&   r'   r(   r5   rI   r*   r+   Ú__classcell__©r>   r"   s   @@r   r-   r-      s    ù‡ € ÷Tó T÷!÷ !ò !r   r-   c                   ó    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V3R lR llt]R 4       t	]R	 4       t
]R
 4       tRtVtV ;t# )r   c                ó    <€ V ^8„  d   QhRS[ /# r/   r   )r!   r"   s   "€r   r#   ÚModel.__annotate__+   s   ø€ ÷ @ñ @™yñ @r   c                ó†   <€ \         SV `  4        Wn        VP                  V n        \	        VP
                  4      V n        R # r   )r4   r5   r6   r   r-   r   Úlanguage_modelr=   s   &&€r   r5   ÚModel.__init__+   s3   ø€ Ü‰ÑÔØŒ	Ø ×+Ñ+ˆŒÜ+¨F×,>Ñ,>Ó?ˆÖr   c                óJ   <€ V ^8„  d   QhRS[ P                  RS[S[,          /# rA   rD   )r!   r"   s   "€r   r#   rO   1   s'   ø€ ÷ 2ñ 2á—‘ð2ñ ™�}ñ2r   c                ó$   € V P                  W4      # r   )rQ   )r   rB   rC   s   &&&r   rI   ÚModel.__call__1   s   € ð
 ×"Ñ" 6Ó1Ð1r   c                óÀ  € \        \        VP                  4       4      4      pVP                  R R4       VP                  RR4       VP                  RR4       VP                  RR4       \	        \        VR,          4      4      p\        P                  ! V P                  V4      p\        \        VP                  4       4      4      VR&   \	        \        V4      4      # )Úvision_towerNÚvision_modelÚmulti_modal_projectorÚmm_projectorrQ   )	r   ÚlistÚitemsÚpopr   r   ÚDeepseekV3LMÚsanitizerQ   )r   ÚweightsÚ
lm_weightss   && r   r_   ÚModel.sanitize8   s«   € Ü ¤ g§m¡m£oÓ!6Ó7ˆØ�‰�N DÔ)Ø�‰�N DÔ)Ø�‰Ð+¨TÔ2Ø�‰�N DÔ)Üœ, wÐ/?Õ'@ÓAÓBˆ
Ü!×*Ò*¨4×+>Ñ+>À
ÓKˆ
Ü$2´4¸
×8HÑ8HÓ8JÓ3KÓ$LˆÐ Ñ!Ü”L Ó)Ó*Ð*r   c                óX   <€ V ^8„  d   QhRS[ S[P                  P                  ,          /# )r   Úgroup)r   rE   ÚdistributedÚGroup)r!   r"   s   "€r   r#   rO   C   s#   ø€ ÷ 7ñ 7™8¡B§N¡N×$8Ñ$8Õ9ñ 7r   c                óH   € \         P                  ! V P                  V4       R # r   )r^   ÚshardrQ   )r   rd   s   &&r   rh   ÚModel.shardC   s   € Ü×Ò˜4×.Ñ.°Ö6r   c                ó.   € V P                   P                  # r   )rQ   r7   r   s   &r   r7   ÚModel.modelF   s   € à×"Ñ"×(Ñ(Ð(r   c                óB   € V P                   P                  P                  # r   )rQ   r7   Úpipeline_layersr   s   &r   ÚlayersÚModel.layersJ   s   € à×"Ñ"×(Ñ(×8Ñ8Ð8r   c                ó   € R  pV# )c                 ó   € R V 9  # )Úe_score_correction_bias© )Úks   &r   Ú	predicateÚ'Model.cast_predicate.<locals>.predicateP   s   € Ø,°AÑ5Ð5r   rs   )r   ru   s   & r   Úcast_predicateÚModel.cast_predicateN   s   € ò	6ð Ðr   )r6   rQ   r   r   )r%   r&   r'   r(   r5   rI   r_   rh   Úpropertyr7   rn   rw   r*   r+   rK   rL   s   @@r   r   r   *   sk   ù‡ € ÷@ó @÷2ò 2ò	+÷7ò 7ð ñ)ó ð)ð ñ9ó ð9ð ñó ÷ð r   r   )Údataclassesr   Útypingr   r   r   Úmlx.coreÚcorerE   Úmlx.nnr8   Ú	mlx.utilsr   r   Úbaser	   Údeepseek_v3r
   r   r^   r   r   ÚModuler-   rs   r   r   Ú<module>rƒ      sc   ðõ "ß 'Ñ 'å Ý ß 2å Ý (Ý .Ý 0ð ôF�ó Fó ðFô!�B—I‘Iô !ô )ˆB�I‰Iö )r   