+
    UV-jÁ	  ã                  ó  a € 0 t $ ^ RIHt ^ RIt^ RIHt ^ RIHt ]! RRR7       ! R R4      4       t]! RRR7       ! R	 R
4      4       t	R]	! ^0RRRR7      R]	! ^RRRR7      R]	! ^RRRR7      /t
R]R&   R R ltRR R lltR R ltR# )é    )ÚannotationsN)Ú	dataclass)Ú
NormalDistT)ÚfrozenÚslotsc                  óX   € ] tR t^t$ R]R&   RtR]R&   RtR]R&   RtR]R&   R R	 ltR
t	R# )ÚLogitNormalScheduleÚfloatÚmeanç      ð?ÚstdÚ
logsnr_ming      2@Ú
logsnr_maxc               ó    € V ^8„  d   QhRRRR/# )é   Útr
   Úreturn© )Úformats   "Ús/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_vlm/models/ideogram4/scheduler.pyÚ__annotate__Ú LogitNormalSchedule.__annotate__   s   € ÷ 
-ñ 
-˜%ð 
- Eñ 
-ó    c                	óö  € R ^\         P                  ! RV P                  ,          4      ,           ,          pR ^\         P                  ! RV P                  ,          4      ,           ,          pVR8:  d   V# VR 8¼  d   V# \	        4       P                  V4      pV P                  V P                  V,          ,           pR R R \         P                  ! V) 4      ,           ,          ,
          p\        \        Wb4      V4      # )r   ç      à?ç        )
ÚmathÚexpr   r   r   Úinv_cdfr   r   ÚminÚmax)Úselfr   Út_minÚt_maxÚzÚshiftedÚvalues   &&     r   Ú__call__ÚLogitNormalSchedule.__call__   sµ   € Ø�qœ4Ÿ8š8 C¨$¯/©/Õ$9Ó:Õ:Õ;ˆØ�qœ4Ÿ8š8 C¨$¯/©/Õ$9Ó:Õ:Õ;ˆØ�Œ8ØˆLØ�Œ8ØˆLÜ‹L× Ñ  Ó#ˆØ—)‘)˜dŸh™h¨�lÕ*ˆØ�s˜c¤D§H¢H¨g¨XÓ$6Õ6Õ7Õ8ˆÜ”3�uÓ$ eÓ,Ð,r   r   Ng      .À)
Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__annotations__r   r   r   r(   Ú__static_attributes__r   r   r   r	   r	      s/   ‡ à
ƒKØ€CˆÓØ€J�ÓØ€J�Ó÷
-ñ 
-r   r	   c                  óP   € ] tR t^t$ R]R&   R]R&   R]R&   RtR]R&   R	 R
 ltRtR# )ÚSamplerPresetÚintÚ	num_stepsútuple[float, ...]Úguidance_scheduler
   Úmur   r   c               ó   € V ^8„  d   QhRR/# )r   r   ÚNoner   )r   s   "r   r   ÚSamplerPreset.__annotate__#   s   € ÷ Nñ N˜tñ Nr   c                	ód   € \        V P                  4      V P                  8w  d   \        R 4      hR# )z-guidance_schedule length must match num_stepsN)Úlenr5   r3   Ú
ValueError)r"   s   &r   Ú__post_init__ÚSamplerPreset.__post_init__#   s*   € Üˆt×%Ñ%Ó&¨$¯.©.Ô8ÜÐLÓMÐMñ 9r   r   N)r*   r+   r,   r-   r.   r   r=   r/   r   r   r   r1   r1      s(   ‡ àƒNØ(Ó(ØƒIØ€CˆÓ÷Nñ Nr   r1   ÚV4_QUALITY_48r   g      ø?)r3   r5   r6   r   ÚV4_DEFAULT_20g      ü?ÚV4_TURBO_12r   zdict[str, SamplerPreset]ÚPRESETSc               ó    € V ^8„  d   QhRRRR/# )r   Únamez
str | Noner   r1   r   )r   s   "r   r   r   >   s   € ÷ ñ �Zð  Mñ r   c           	     óœ   € T ;'       g    R p \         V,          #   \         d&   p\        RT : R\        \         4       24      ThRp?ii ; i)r@   z"Unknown Ideogram 4 sampler preset z; expected one of N)rB   ÚKeyErrorr<   Úsorted)rD   ÚkeyÚexcs   &  r   Ú
get_presetrJ   >   s[   € Ø
×
!Ð
!�/€CðÜ�s�|ÐøÜô ÜØ0°±ð 9Ü%¤g›Ð/ð1ó
ð ð	ûðús   Ž ›A¦ AÁAc          
     ó,   € V ^8„  d   QhRRRRRRRRRR/# )	r   Úimage_resolutionztuple[int, int]Úknown_resolutionÚ
known_meanr
   r   r   r	   r   )r   s   "r   r   r   I   s:   € ÷ 	3ñ 	3Ø%ð	3à%ð	3ð ð	3ð 
ð		3ð
 ñ	3r   c                óÊ   € V ^ ,          V ^,          ,          pV^ ,          V^,          ,          pVR\         P                  ! WE,          4      ,          ,           p\        WcR7      # )r   r   )r   r   )r   Úlogr	   )rL   rM   rN   r   Ú
num_pixelsÚknown_pixelsr   s   &&&&   r   Úget_schedule_for_resolutionrS   I   sT   € ð " !Õ$Ð'7¸Õ':Õ:€JØ# AÕ&Ð)9¸!Õ)<Õ<€LØ˜œdŸhšh zÕ'@ÓAÕAÕA€DÜ DÔ2Ð2r   c               ó    € V ^8„  d   QhRRRR/# )r   r3   r2   r   r4   r   )r   s   "r   r   r   U   s   € ÷ >ñ > 3ð >Ð+<ñ >r   c                óÒ   a € S ^8  d   \        RS  24      h\        ;QJ d'    . V 3R l\        S ^,           4       4       F  NK  	  5# ! V 3R l\        S ^,           4       4       4      # )é   znum_steps must be >= 1, got c              3  ó4   <"  € T F  qS,          x € K  	  R # 5i)Nr   )Ú.0Úir3   s   & €r   Ú	<genexpr>Ú&make_step_intervals.<locals>.<genexpr>X   s   øé € Ð=Ñ(< 1�Y—’Ó(<ùs   ƒ)r<   ÚtupleÚrange)r3   s   fr   Úmake_step_intervalsr^   U   sQ   ø€ Ø�1„}ÜÐ7¸	°{ÐCÓDÐDßŒ5Ô=¬¨i¸!­mÔ(<Ó=�5Ð=ˆ5Ô=¬¨i¸!­mÔ(<Ó=Ó=Ð=r   )0ç      @r_   r_   ç      @r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   )r_   r_   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   )r_   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   r`   ))é   ra   r   r   )Ú__conditional_annotations__Ú
__future__r   r   Údataclassesr   Ú
statisticsr   r	   r1   rB   r.   rJ   rS   r^   )rb   s   @r   Ú<module>rf      s¾   øðß "Ð "ã Ý !Ý !ñ �$˜dÔ#÷-ð -ó $ð-ñ& �$˜dÔ#÷Nð Nó $ðNð ‘]ØØ2ØØô	ð ‘]ØØ2ØØô	ð ‘=ØØ2ØØô	ð%€Ð	!ó õ,÷	3÷>r   