+
    UV-j;  ã                  ó†   € ^ RI Ht ^ RIt^ RIHt ^ RIHt ^ RIHtH	t	 Rt
RRRRRRRR	R
R^/R R llltR R ltR R ltR# )é    )ÚannotationsN)ÚPath)Úsnapshot_download)ÚFlux2VariantÚget_variantÚ	local_dirÚtokenÚrevisionÚforce_downloadFÚmax_workersc               ó4   € V ^8„  d   QhRRRRRRRRRR	R
RRR/# )é   Úvariantústr | Flux2Variantr   zstr | Path | Noner	   z
str | Noner
   r   Úboolr   ÚintÚreturnr   © )Úformats   "Ún/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_vlm/models/flux2/download.pyÚ__annotate__r      sP   € ÷ #ñ #Øð#ð !ð#ð ð	#ð
 ð#ð ð#ð ð#ð 
ñ#ó    c               ó   € \        V 4      pVf   V'       g   \        V4      pVe   V# RVP                  RTR\        \        4      RT;'       g*    \
        P                  P                  R4      ;'       g    R RVRV/pVe;   \        V4      P                  4       p	V	P                  RRR	7       \        V	4      VR
&    \        \        R/ VB 4      p
\        V
4      #   \         d    \        T4      pTe   Tu # h i ; i)NÚrepo_idr
   Úallow_patternsr	   ÚHF_TOKENr   r   T)ÚparentsÚexist_okr   r   )r   Úfind_valid_cached_snapshotr   ÚlistÚDOWNLOAD_PATTERNSÚosÚenvironÚgetr   Ú
expanduserÚmkdirÚstrr   Úvalidate_model_layoutÚFileNotFoundError)r   r   r	   r
   r   r   ÚspecÚcachedÚkwargsÚtargetÚ
model_paths   &$$$$$     r   Údownload_modelr/      sû   € ô �wÓ€DØÒ§Ü+¨DÓ1ˆØÒØˆMð 	�4—<‘<Ø�HØœ$Ô0Ó1Ø�×<Ð<œ"Ÿ*™*Ÿ.™.¨Ó4×<Ð<¸Ø˜.Ø�{ð€Fð ÒÜ�i“×+Ñ+Ó-ˆØ�‰˜T¨DˆÔ1Ü! &›kˆˆ{ÑðÜÔ+Ñ5¨fÑ5Ó6ˆ
Ü$ ZÓ0Ð0øÜô Ü+¨DÓ1ˆØÒØŠMØð	ús   Â>C ÃC=Ã;C=c               ó    € V ^8„  d   QhRRRR/# )r   r   r   r   zPath | Noner   )r   s   "r   r   r   >   s   € ÷ ñ Ð(:ð ¸{ñ r   c                ó‚  € \        V 4      p\        P                  ! 4       R ,          R,          R,          pVRVP                  P	                  RR4       2,          pVR,          pVP                  4       '       g   R# \        VP                  4       R R	R
7       F  p \        V4      u # 	  R#   \         d     K#  i ; i)z.cacheÚhuggingfaceÚhubzmodels--Ú/z--Ú	snapshotsNc                ó6   € V P                  4       P                  # )N)ÚstatÚst_mtime)Úps   &r   Ú<lambda>Ú,find_valid_cached_snapshot.<locals>.<lambda>F   s   € ¨1¯6©6«8×+<Ò+<r   T)ÚkeyÚreverse)
r   r   Úhomer   ÚreplaceÚexistsÚsortedÚiterdirr(   r)   )r   r*   Ú
cache_rootÚrepo_dirr5   Úsnapshots   &     r   r   r   >   s®   € Ü�wÓ€DÜ—’“˜xÕ'¨-Õ7¸%Õ?€JØ˜h t§|¡|×';Ñ';¸CÀÓ'FÐ&GÐHÕH€HØ˜;Õ&€IØ×Ñ×ÒÙÜØ×ÑÓÑ!<Àd÷ˆð	Ü(¨Ó2Ò2ñ	ñ øô !ô 	Úð	ús   Â
B/Â/B>Â=B>c               ó    € V ^8„  d   QhRRRR/# )r   r.   z
str | Pathr   r   r   )r   s   "r   r   r   O   s   € ÷ ñ  jð °Tñ r   c                óP  € \        V 4      P                  4       pVP                  4       '       g   \        R V 24      hRpV Uu. uF'  p\	        VP                  V4      4      '       d   K%  VNK)  	  ppV'       d'   RP                  R V 4       4      p\        RV 24      hV# u upi )zModel path does not exist: Ú
c              3  ó,   "  € T F
  pR V 2x € K  	  R# 5i)z  - Nr   )Ú.0Úitems   & r   Ú	<genexpr>Ú(validate_model_layout.<locals>.<genexpr>\   s   é € Ð@¹°  T F�m»ùs   ‚z0Model snapshot is missing required Flux2 files:
)útransformer/*.safetensorsútext_encoder/*.safetensorsúvae/*.safetensorsztokenizer/tokenizer.json)r   r%   r@   r)   r    ÚglobÚjoin)r.   ÚrootÚrequired_globsÚpatternÚmissingÚ	formatteds   &     r   r(   r(   O   s�   € Ü�
Ó×&Ñ&Ó(€DØ�;‰;�=Š=ÜÐ"=¸d¸VÐ DÓEÐEð€Nñ '5ÓU¡n˜7¼DÀÇÁÈ7ÓAS×<T�wˆw¡n€GÐUßØ—I‘IÑ@¹Ó@Ó@ˆ	ÜØ?À	¸{ÐKó
ð 	
ð €Kùò Vs   Á"B#Á+B#)
zmodel_index.jsonrP   z
vae/*.jsonrN   ztransformer/*.jsonrO   ztext_encoder/*.jsonztokenizer/**zadded_tokens.jsonzchat_template.jinja)zflux2-klein-4b)Ú
__future__r   r"   Úpathlibr   Úhuggingface_hubr   Úmlx_vlm.models.flux2.configr   r   r!   r/   r   r(   r   r   r   Ú<module>r\      sc   ðÝ "ã 	Ý å -ç AðÐ ñ#ð $(ð#ð ð	#ð
  ð#ð !ð#ð ÷#ð #õL÷"r   