+
    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HtHt ^RIHtHtHtHt ^RIHt ^R	IHt R
 R ltR R lt]! RR7       ! R R]4      4       tRR R llt. ROtR# )é    )Úannotations)Ú	dataclass)ÚPath)ÚAnyÚClassVarN)ÚImageGenerationModelÚImageGenerationRequestÚImageGenerationResult)ÚIDEOGRAM_4_FP8_REPO_IDÚIdeogram4VariantÚget_variantÚvariant_from_local_path)Úvalidate_model_layout)ÚIdeogram4ImagePipelinec               ó    € V ^8„  d   QhRRRR/# )é   Úmodelzstr | Ideogram4Variant | NoneÚreturnr   © )Úformats   "Úo/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_vlm/models/ideogram4/model.pyÚ__annotate__r      s   € ÷ #ñ #Ð8ð #Ð=Mñ #ó    c                ó  € \        V \        4      '       d   V # V f   \        \        4      # \	        \        V 4      4      P                  4       pVP                  4       '       d   \        V4      # \        \        V 4      4      # ©N)	Ú
isinstancer   r   r   r   ÚstrÚ
expanduserÚexistsr   ©r   Ú
model_paths   & r   Úresolve_variantr"      sf   € Ü�%Ô)×*Ò*ØˆØ‚}ÜÔ1Ó2Ð2Ü”c˜%“jÓ!×,Ñ,Ó.€JØ×Ñ×ÒÜ& zÓ2Ð2Ü”s˜5“zÓ"Ð"r   c               ó    € V ^8„  d   QhRRRR/# ©r   r   r   r   Úboolr   )r   s   "r   r   r   $   s   € ÷ ñ �Cð ˜Dñ r   c                óì   € \        V 4      P                  4       pVP                  4       '       d    \        V4       R #  \        V 4       R #   \        \
        3 d     R# i ; i  \
         d     R# i ; i)TF)r   r   r   r   ÚFileNotFoundErrorÚ
ValueErrorr"   r    s   & r   Úcan_loadr)   $   so   € Ü�e“×'Ñ'Ó)€JØ×Ñ×Òð	Ü! *Ô-ÙðÜ˜ÔÙøô	 "¤:Ð.ô 	Úð	ûô
 ô Úðús"   ±A ¿A$ ÁA!Á A!Á$A3Á2A3T)Úslotsc                  ó¨   € ] tR t^3t$ RtR]R&   RtR]R&   R]R&   R	]R
&   RtR	]R&   ]R R l4       t	R R lt
]R R l4       t]RR R ll4       tRtR# )ÚIdeogram4ImageGenerationModelTzClassVar[bool]Úis_image_generation_modelÚ	ideogram4zClassVar[str]Ú
model_typer   Úpipeliner   Úmodel_idÚfamilyc               ó   € V ^8„  d   QhRR/# )r   r   r   r   )r   s   "r   r   Ú*Ideogram4ImageGenerationModel.__annotate__<   s   € ÷ *ñ *˜ñ *r   c                	óB   € V P                   P                  P                  # r   )r0   ÚvariantÚname)Úselfs   &r   r6   Ú%Ideogram4ImageGenerationModel.variant;   s   € à�}‰}×$Ñ$×)Ñ)Ð)r   c               ó    € V ^8„  d   QhRRRR/# )r   Úrequestr	   r   r
   r   )r   s   "r   r   r4   ?   s   € ÷ 
ñ 
Ð 6ð 
Ð;Pñ 
r   c                	ót  € VP                   f   ^ MVP                   pV P                  P                  ! VP                  3RVRVP                  RVP
                  RVP                  RVP                  /VP                  B w  r4\        VVVP
                  VP                  \        VP                  RVP                  4      4      V P                  V P                  V P                  \        VP                  RVP                  4      4      VP                  R4      \         P"                  ! 4       R,          VR7      # )	NÚseedÚstepsÚwidthÚheightÚguidanceÚprompt_tokensg    eÍÍA)Úarrayr=   r?   r@   r>   r   r2   r6   rA   rB   Úpeak_memoryÚmetadata)r=   r0   Úgenerate_arrayÚpromptr>   r?   r@   rA   Úextrar
   ÚintÚgetr1   r2   r6   ÚfloatÚmxÚget_peak_memory)r8   r;   r=   rC   rE   s   &&   r   ÚgenerateÚ&Ideogram4ImageGenerationModel.generate?   s  € Ø—L‘LÒ(‰q¨g¯l©lˆØŸ-™-×6Ò6Ø�N‰Nñ
àð
ð —-‘-ð
ð —-‘-ð	
ð
 —>‘>ð
ð ×%Ñ%ð
ð �m‰mñ
‰ˆô %ØØØ—-‘-Ø—>‘>Ü�h—l‘l 7¨G¯M©MÓ:Ó;Ø—-‘-Ø—;‘;Ø—L‘LÜ˜8Ÿ<™<¨
°G×4DÑ4DÓEÓFØ"Ÿ,™, Ó7Ü×*Ò*Ó,¨sÕ2Øô
ð 	
r   c               ó    € V ^8„  d   QhRRRR/# r$   r   )r   s   "r   r   r4   Z   s   € ÷ ñ  3ð ¨4ñ r   c                	ó   € \        V4      # r   )r)   )Úclsr   s   &&r   Úsupports_modelÚ,Ideogram4ImageGenerationModel.supports_modelY   s   € ä˜‹Ðr   c               ó$   € V ^8„  d   QhRRRRRR/# )r   r   r   Úkwargsr   r   z'Ideogram4ImageGenerationModel'r   )r   s   "r   r   r4   ^   s(   € ÷ ;ñ ;àð;ð ð;ð 
)ñ	;r   c                	óÚ  € VP                  R R4      pVf]   \        V\        4      '       dG   \        V4      P	                  4       P                  4       '       d   \        V4      P	                  4       MTp\        P                  ! \        V4      VVP                  RR4      VP                  RR4      VP                  RR4      VP                  RR4      R7      pV ! V\        V4      R	7      # )
r!   NÚdownloadTÚtokenÚrevisionÚforce_downloadF)r!   rX   rY   rZ   r[   )r0   r1   )	Úpopr   r   r   r   r   r   Úfrom_pretrainedr"   )rR   r   rV   Úmodel_path_argr!   r0   s   &&,   r   Úfrom_model_idÚ+Ideogram4ImageGenerationModel.from_model_id]   sÊ   € ð  Ÿ™ L°$Ó7ˆð Ò&Ü˜u¤c×*Ò*Ü˜“K×*Ñ*Ó,×3Ñ3×5Ò5ô	 �‹K×"Ñ"Ô$ð  ð 	ô *×9Ò9Ü˜EÓ"Ø!Ø—Z‘Z 
¨DÓ1Ø—*‘*˜W dÓ+Ø—Z‘Z 
¨DÓ1Ø!Ÿ:™:Ð&6¸Ó>ô
ˆñ ˜H¬s°5«zÔ:Ð:r   r   N©zideogram-ai/ideogram-4-fp8)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r-   Ú__annotations__r/   r2   Úpropertyr6   rN   ÚclassmethodrS   r_   Ú__static_attributes__r   r   r   r,   r,   3   sk   ‡ à04Ð˜~Ó4Ø +€J�Ó+Ø$Ó$ØƒMØ€FˆCÓàô*ó ð*õ
ð4 ôó ðð ö;ó ô;r   r,   c               ó$   € V ^8„  d   QhRRRRRR/# )r   r   r   rV   r   r   r,   r   )r   s   "r   r   r   x   s)   € ÷ Hñ HØðHØ9<ðHà"ñHr   c                ó0   € \         P                  ! V 3/ VB # r   )r,   r_   )r   rV   s   &,r   Úloadrl   x   s   € ô )×6Ò6°uÑGÀÑGÐGr   ra   )r,   r)   rl   r"   )Ú
__future__r   Údataclassesr   Úpathlibr   Útypingr   r   Úmlx.coreÚcorerL   Úmlx_vlm.generate.imager   r	   r
   Úconfigr   r   r   r   rX   r   r0   r   r"   r)   r,   rl   Ú__all__r   r   r   Ú<module>rv      sl   ðÝ "å !Ý ß  å ÷ñ ÷ó õ ,Ý ,õ#õñ �ÔôA;Ð$8ó A;ó ðA;÷HHò‚r   