+
    UV-j“  ã                  ó”   € ^ RI Ht ^ RIHt ^ RIHt ^ RIHt ^ RI	t
]! RRR7       ! R R4      4       tR	RR
RR^/R R lltR R ltR# )é    )Úannotations)Ú	dataclass)ÚCallableNT)ÚfrozenÚslotsc                  óB   € ] tR t^
t$ ^tR]R&   ^€tR]R&   ^tR]R&   RtR# )ÚTilingConfigz
int | NoneÚvae_decode_tiles_per_dimÚintÚvae_decode_tile_sizeÚvae_decode_overlap© N)	Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r
   Ú__annotations__r   r   Ú__static_attributes__r   ó    Úl/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_vlm/models/flux2/tiling.pyr	   r	   
   s#   ‡ à+,Ð˜jÓ,Ø #Ð˜#Ó#ØÐ˜×r   r	   Ú	tile_sizeÚtile_overlapÚspatial_scalec               ó0   € V ^8„  d   QhRRRRRRRRRR	R
R/# )é   Úlatentzmx.arrayÚ	decode_fnzCallable[[mx.array], mx.array]r   ztuple[int, int]r   r   r   Úreturnr   )Úformats   "r   Ú__annotate__r       sN   € ÷ X:ñ X:àðX:ð .ðX:ð ð	X:ð
 "ðX:ð ðX:ð ñX:r   c           
     ó
  € V P                   w  rVrxp	V^8w  g   V^8w  d,   V! V 4      p
V
P                   ^,          ^8X  d
   V
R,          # T
# \        V4      pW‹,          pW›,          pVw  rïVw  pp\        ^Wë,          4      p\        ^Wû,          4      pVV8:  d3   V	V8:  d,   V! V 4      p
V
P                   ^,          ^8X  d
   V
R,          # T
# \        ^ \        VV,          V^,
          4      4      p\        ^ \        VV,          V^,
          4      4      p\        ^VV,
          4      p\        ^VV,
          4      p\	        V4      p\	        V4      pRpRp\        ^ VV4       EFE  p\        VV,           V4      p\        ^ V	V4       EF  p\        VV,           V	4      pV^ 8”  d   VV,
          V8:  g   V^ 8”  d   VV,
          V8:  d   KC  V RRRVV1VV13,          p V! V 4      p!V!P                   ^,          ^8X  d
   V!R,          p!\        P                  ! V!P                  \        P                  4      4      ^ ,          P                  ^^^ 4      p"VV,          p#VV,          p$VV,
          V,          p%VV,
          V,          p&\        V%V"P                   ^ ,          VV#,
          4      p'\        V&V"P                   ^,          VV$,
          4      p(V"RV'1RV(1R3,          p"Vfb   \        P                  ! WÍV"P                   ^,          3\        P                  R7      p\        P                  ! WÍ^3\        P                  R7      p\        ^ \        VV'^,
          4      4      p)\        ^ \        VV(^,
          4      4      p*\        P                  ! V'3\        P                  R7      p+\        P                  ! V(3\        P                  R7      p,V)^ 8”  d'   V^ 8”  d	   VRV) V+RV)% VV8  d   RVRV) ,
          V+V)) R% V*^ 8”  d'   V^ 8”  d	   VRV* V,RV*% VV	8  d   RVRV* ,
          V,V*) R% V+R,          V,R,          ,          p-VV#V#V',           1V$V$V(,           1R3;;,          V"V-R	,          ,          ,          uu&   VV#V#V',           1V$V$V(,           1R3;;,          V-R	,          ,          uu&   EK  	  EKH  	  Ve   Vf   Q hV\        P                  ! VRR4      ,          p\        P                  ! VP                  ^^ ^4      R
,          4      # )é   ºNNNN©Údtypeç      ð?g�íµ ÷Æ°>)r#   r#   r   r#   r#   )r#   N)Nr#   )r#   r#   N)N.)Úshaper   ÚmaxÚminÚ	_cos_rampÚrangeÚnpÚarrayÚastypeÚmxÚfloat32Ú	transposeÚzerosÚonesÚclip).r   r   r   r   r   ÚbatchÚ_ÚframesÚ
height_latÚ	width_latÚdecodedÚscaleÚ
height_outÚ	width_outÚtile_hÚtile_wÚ	overlap_hÚ	overlap_wÚlatent_tile_hÚlatent_tile_wÚlatent_overlap_hÚlatent_overlap_wÚstride_hÚstride_wÚramp_hÚramp_wÚout_npÚcount_npÚy_latÚ	y_lat_endÚx_latÚ	x_lat_endÚtile_latentÚdecoded_tileÚtile_npÚy_outÚx_outÚh_outÚw_outÚeff_hÚeff_wÚov_h_outÚov_w_outÚwhÚwwÚweightss.   $$$$$                                         r   Údecode_image_tiledr^      s¢  € ð /5¯l©lÑ+€Eˆf )Ø�„z�V˜q”[Ù˜FÓ#ˆØ)0¯©°qÕ)9¸QÔ)>ˆw�}Õ%ÐKÀGÐKä�Ó€EØÕ#€JØÕ!€IØ�N€FØ'Ñ€IˆyÜ˜˜6�?Ó+€MÜ˜˜6�?Ó+€Mà�]Ô" y°MÔ'AÙ˜FÓ#ˆØ)0¯©°qÕ)9¸QÔ)>ˆw�}Õ%ÐKÀGÐKä˜1œc )¨uÕ"4°mÀaÕ6GÓHÓIÐÜ˜1œc )¨uÕ"4°mÀaÕ6GÓHÓIÐÜ�1�mÐ&6Õ6Ó7€HÜ�1�mÐ&6Õ6Ó7€HÜ�yÓ!€FÜ�yÓ!€FØ $€FØ"&€Hä�q˜* h×/ˆÜ˜ Õ-¨zÓ:ˆ	Ü˜1˜i¨×2ˆEÜ˜E MÕ1°9Ó=ˆIØ˜”	˜y¨5Õ0Ð5EÔEØ˜”	˜y¨5Õ0Ð5EÔEáà   A q¨%°	¨/¸5À¸?Ð!JÕKˆKÙ$ [Ó1ˆLØ×!Ñ! !Õ$¨Ô)Ø+¨MÕ:�Ü—h’h˜|×2Ñ2´2·:±:Ó>Ó?ÀÕB×LÑLÈQÐPQÐSTÓUˆGà˜E•MˆEØ˜E•MˆEØ Õ&¨%Õ/ˆEØ Õ&¨%Õ/ˆEÜ˜˜wŸ}™}¨QÕ/°¸eÕ1CÓDˆEÜ˜˜wŸ}™}¨QÕ/°¸UÕ1BÓCˆEØ˜f˜u˜f f u f¨aÐ/Õ0ˆGàŠ~ÜŸšØ¨G¯M©M¸!Õ,<Ð=ÄRÇZÁZô�ô Ÿ8š8 Z¸AÐ$>ÄbÇjÁjÔQ�ä˜1œc )¨U°Q­YÓ7Ó8ˆHÜ˜1œc )¨U°Q­YÓ7Ó8ˆHÜ—’˜%˜¬¯©Ô4ˆBÜ—’˜%˜¬¯©Ô4ˆBØ˜!Œ|Ø˜1”9Ø$*¨9¨HÐ$5�B�y˜�MØ˜zÔ)Ø%(¨6°)°8Ð+<Õ%<�B˜�y�z�NØ˜!Œ|Ø˜1”9Ø$*¨9¨HÐ$5�B�y˜�MØ˜yÔ(Ø%(¨6°)°8Ð+<Õ%<�B˜�y�z�Nà˜•k B w¥KÕ/ˆGØ�5˜5 5�=Ð(¨%°%¸%µ-Ð*?ÀÐB×CØ˜' *Õ-Õ-õÓCð �U˜U U�]Ð*¨E°E¸EµMÐ,AÀ1ÐD×EÈØõJõ ÖEô] 3ñ 0ðh Ò (Ò"6Ð6Ð6Ø”b—g’g˜h¨¨dÓ3Õ3€FÜ�8Š8�F×$Ñ$ Q¨¨1Ó-¨iÕ8Ó9Ð9r   c               ó    € V ^8„  d   QhRRRR/# )r   Únr   r   z
np.ndarrayr   )r   s   "r   r    r    l   s   € ÷ )ñ )�ð )˜ñ )r   c                ó  € V ^ 8:  d'   \         P                  ! R\         P                  R7      # \         P                  ! RRV \         P                  R7      pRR\         P                  ! V\         P
                  ,          4      ,          ,
          # )r   r$   g        r&   )Únumr%   g      à?)r   )r,   r2   r0   ÚlinspaceÚcosÚpi)r`   Úts   & r   r*   r*   l   sW   € ØˆA„vÜ�xŠx˜¤B§J¡JÔ/Ð/Ü
�Š�C˜ !¬2¯:©:Ô6€AØ�”r—v’v˜a¤"§%¡%�iÓ(Õ(Õ(Ð(r   )é   rg   )é@   rh   )Ú
__future__r   Údataclassesr   Útypingr   Úmlx.coreÚcorer/   Únumpyr,   r	   r^   r*   r   r   r   Ú<module>ro      sa   ðÝ "å !Ý å Û ñ �$˜dÔ#÷ ð  ó $ð ðX:ð ",ð	X:ð
 %-ðX:ð ÷X:÷v)r   