+
    UV-j°  ã                   ó|   € ^ 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]P                  4      tR# )é    )ÚOptionalN)ÚInputEmbeddingsFeaturesÚLanguageModelOutput©ÚModelConfig)ÚLanguageModelc                   ó¦   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V3R lR lltR t]R 4       t	]R	 4       t
]R
 4       tR tRtVtV ;t# )ÚModelc                ó    <€ V ^8„  d   QhRS[ /# )é   Úconfigr   )ÚformatÚ__classdict__s   "€Úm/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_vlm/models/laguna/laguna.pyÚ__annotate__ÚModel.__annotate__   s   ø€ ÷ 4ñ 4™{ñ 4ó    c                ór   <€ \         SV `  4        Wn        VP                  V n        \	        V4      V n        R # ©N)ÚsuperÚ__init__r   Ú
model_typer   Úlanguage_model)Úselfr   Ú	__class__s   &&€r   r   ÚModel.__init__   s-   ø€ Ü‰ÑÔØŒØ ×+Ñ+ˆŒÜ+¨FÓ3ˆÖr   c                ót   <€ V ^8„  d   QhRS[ S[P                  ,          RS[ S[P                  ,          RS[/# )r   Ú	input_idsÚpixel_valuesÚreturn)r   ÚmxÚarrayr   )r   r   s   "€r   r   r      s;   ø€ ÷ 
ñ 
á™BŸH™HÕ%ð
ñ ™rŸx™xÕ(ð
ñ
 
!ñ
r   c                ó`   € \        V P                  P                  P                  V4      R 7      # ))Úinputs_embeds)r   r   ÚmodelÚembed_tokens)r   r   r   Úkwargss   &&&,r   Úget_input_embeddingsÚModel.get_input_embeddings   s+   € ô 'Ø×-Ñ-×3Ñ3×@Ñ@ÀÓKô
ð 	
r   c                ón   <€ V ^8„  d   QhRS[ P                  RS[ P                  RS[ P                  RS[/# )r   r   r   Úmaskr    )r!   r"   r   )r   r   s   "€r   r   r      s?   ø€ ÷ 
ñ 
á—8‘8ð
ñ —h‘hð
ñ �h‰hð	
ñ 
ñ
r   c                óf   € V P                  W4      pV P                  ! V3R VRVP                  /VB # )Úcacher$   )r(   r   r$   )r   r   r   r+   r-   r'   Úinput_embeddings_featuress   &&&&&, r   Ú__call__ÚModel.__call__   sM   € ð %)×$=Ñ$=¸iÓ$VÐ!Ø×"Ò"Øñ
àð
ð 4×AÑAð
ð ñ	
ð 	
r   c                ó    € V P                   P                  V4      pR  pVP                  4        UUu/ uF  w  r4V! V4      VbK  	  upp# u uppi )c                 óž   € V P                  R 4      '       d   V # V P                  R4      '       g   V P                  R4      '       d   R V  2# V # )zlanguage_model.zmodel.zlm_head.)Ú
startswith)Úkeys   &r   Útransform_keyÚ%Model.sanitize.<locals>.transform_key/   sF   € Ø�~‰~Ð/×0Ò0Ø�
Ø�~‰~˜h×'Ò'¨3¯>©>¸*×+EÒ+EØ(¨¨Ð.Ð.ØˆJr   )r   ÚsanitizeÚitems)r   Úweightsr5   ÚkÚvs   &&   r   r7   ÚModel.sanitize,   sI   € Ø×%Ñ%×.Ñ.¨wÓ7ˆò	ð 18·±´Ô@±©¨‘˜aÓ  !Ò#±Ò@Ð@ùÓ@s   ²A
c                ó.   € V P                   P                  # r   )r   Úquant_predicate©r   s   &r   r>   ÚModel.quant_predicate8   s   € à×"Ñ"×2Ñ2Ð2r   c                ó.   € V P                   P                  # r   )r   Úcast_predicater?   s   &r   rB   ÚModel.cast_predicate<   s   € à×"Ñ"×1Ñ1Ð1r   c                ó.   € V P                   P                  # r   )r   Úlayersr?   s   &r   rE   ÚModel.layers@   s   € à×"Ñ"×)Ñ)Ð)r   c                ó6   € V P                   P                  4       # r   )r   Ú
make_cacher?   s   &r   rH   ÚModel.make_cacheD   s   € Ø×"Ñ"×-Ñ-Ó/Ð/r   )r   r   r   )NN)NNN)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r   r(   r/   r7   Úpropertyr>   rB   rE   rH   Ú__static_attributes__Ú__classdictcell__Ú__classcell__)r   r   s   @@r   r
   r
      so   ù‡ € ÷4ó 4÷
ò 
÷
ò 
ò 
Að ñ3ó ð3ð ñ2ó ð2ð ñ*ó ð*÷0ò 0r   r
   )Útypingr   Úmlx.coreÚcorer!   Úmlx.nnÚnnÚbaser   r   r   r   Úlanguager   ÚModuler
   © r   r   Ú<module>r[      s'   ðÝ å Ý ç ?Ý Ý #ô:0ˆB�I‰Iö :0r   