+
    UV-j#  ã                  óð   a € 0 t $ ^ RIHt ^ RIt^ RIH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t]]
! R]R7      /tR]R&   RR R lltR R ltR R ltR# )é    )ÚannotationsN)Ú	dataclass)ÚPath)ÚAnyT)ÚfrozenÚslotsc                  ó:   € ] tR t^	t$ R]R&   R]R&   RtR]R&   RtR# )ÚIdeogram4VariantÚstrÚnameÚrepo_idÚV4_DEFAULT_20Údefault_sampler_preset© N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__annotations__r   Ú__static_attributes__r   ó    Úp/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_vlm/models/ideogram4/config.pyr
   r
   	   s   ‡ à
ƒIØƒLØ"1Ð˜C×1r   r
   c                  óº   € ] tR t^t$ RtR]R&   ^"tR]R&   ^tR]R&   RtR]R&   Rt	R]R	&   ^€t
R]R
&   RtR]R&   RtR]R&   RtR]R&   RtR]R&   ]R R l4       tRtR# )ÚIdeogram4TransformerConfigi   ÚintÚemb_dimÚ
num_layersÚ	num_headsi 0  Úintermediate_sizei   Ú
adanln_dimÚin_channelsÚllm_features_dimi@KL Ú
rope_thetaztuple[int, int, int]Úmrope_sectiongñhãˆµøä>ÚfloatÚnorm_epsc               ó    € V ^8„  d   QhRRRR/# )é   Údatazdict[str, Any]Úreturnz'Ideogram4TransformerConfig'r   )Úformats   "r   Ú__annotate__Ú'Ideogram4TransformerConfig.__annotate__   s   € ÷ 
ñ 
˜^ð 
Ð0Lñ 
r   c                	ó„  € VP                  R VP                  RV P                  4      4      pVP                  R4      pVP                  R4      pVf    Ve   \        V4      \        V4      ,          pT ! \        T;'       g    V P                  4      \        VP                  RV P                  4      4      \        V4      \        VP                  RV P
                  4      4      \        VP                  RV P                  4      4      \        VP                  RV P                  4      4      \        VP                  RV P                  4      4      \        VP                  R	V P                  4      4      \        VP                  R
V P                  4      4      \        VP                  RV P                  4      4      R7
      # )r   Únum_attention_headsÚattention_head_dimr   r   r   Ú	adaln_dimr!   r"   r#   r$   r&   )
r   r   r   r   r    r!   r"   r#   r$   r&   )Úgetr   r   r   r   r   r    r!   r"   r#   Útupler$   r%   r&   )Úclsr)   ÚheadsÚhead_dimr   s   &&   r   Ú	from_dictÚ$Ideogram4TransformerConfig.from_dict   sC  € à—‘˜ d§h¡hÐ/DÀcÇmÁmÓ&TÓUˆØ—8‘8Ð0Ó1ˆØ—(‘(˜9Ó%ˆØŠ?˜xÒ3Ü˜%“j¤3 x£=Õ0ˆGÙÜ˜×.Ð. 3§;¡;Ó/Ü˜4Ÿ8™8 L°#·.±.ÓAÓBÜ˜%“jÜ! $§(¡(Ð+>À×@UÑ@UÓ"VÓWÜ˜4Ÿ8™8 K°·±Ó@ÓAÜ˜DŸH™H ]°C·O±OÓDÓEÜ  §¡Ð*<¸c×>RÑ>RÓ!SÓTÜ˜4Ÿ8™8 L°#·.±.ÓAÓBÜ §¡¨¸#×:KÑ:KÓ LÓMÜ˜4Ÿ8™8 J°·±Ó=Ó>ô
ð 	
r   r   Ni Ð  )é   é   r:   )r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r&   Úclassmethodr7   r   r   r   r   r   r      s{   ‡ à€GˆSÓØ€J�ÓØ€IˆsÓØ"Ð�sÓ"Ø€J�ÓØ€K�ÓØ%Ð�cÓ%Ø€J�ÓØ*6€MÐ'Ó6Ø€HˆeÓàô
ó ô
r   r   zideogram-ai/ideogram-4-fp8zideogram-4-fp8)r   r   zdict[str, Ideogram4Variant]ÚVARIANTSc               ó    € V ^8„  d   QhRRRR/# )r(   Úmodelzstr | Ideogram4Variant | Noner*   r
   r   )r+   s   "r   r,   r,   <   s    € ÷ 	Nñ 	NÐ4ð 	NÐ@Pñ 	Nr   c                ó(  € \        V \        4      '       d   V # V f   \        \        ,          # \	        V 4      P                  4       P                  4       P                  R4      p \        V,          #   \         d   p\        RT  24      ThR p?ii ; i)NÚ/z Unsupported Ideogram 4 variant: )
Ú
isinstancer
   r<   ÚIDEOGRAM_4_FP8_REPO_IDr   ÚstripÚlowerÚrstripÚKeyErrorÚ
ValueError)r>   ÚkeyÚexcs   &  r   Úget_variantrJ   <   sƒ   € Ü�%Ô)×*Ò*ØˆØ‚}ÜÔ.Õ/Ð/Ü
ˆe‹*×
Ñ
Ó
×
"Ñ
"Ó
$×
+Ñ
+¨CÓ
0€CðNÜ˜�}ÐøÜô NÜÐ;¸E¸7ÐCÓDÈ#ÐMûðNús   Á%A2 Á2BÁ=BÂBc               ó    € V ^8„  d   QhRRRR/# )r(   Ú
model_pathz
str | Pathr*   r
   r   )r+   s   "r   r,   r,   H   s   € ÷ /ñ /¨
ð /Ð7Gñ /r   c                ó<  € \        V 4      P                  4       pVR ,          pVP                  4       '       g   \        RV 24      h\        P
                  ! VP                  4       4      pVP                  R4      R8w  d   \        V R24      h\        \        4      # )zmodel_index.jsonz*Missing Ideogram 4 model_index.json under Ú_class_nameÚIdeogram4Pipelinez% is not an Ideogram4Pipeline snapshot)r   Ú
expanduserÚexistsÚFileNotFoundErrorÚjsonÚloadsÚ	read_textr2   rG   rJ   rB   )rL   ÚrootÚ
index_pathr)   s   &   r   Úvariant_from_local_pathrX   H   sŠ   € Ü�
Ó×&Ñ&Ó(€DØÐ*Õ*€JØ×Ñ×ÒÜÐ"LÈTÈFÐ SÓTÐTÜ�:Š:�j×*Ñ*Ó,Ó-€DØ‡x�x�ÓÐ"5Ô5Ü˜D˜6Ð!FÐGÓHÐHÜÔ-Ó.Ð.r   c               ó$   € V ^8„  d   QhRRRRRR/# )r(   Úwidthr   Úheightr*   ÚNoner   )r+   s   "r   r,   r,   S   s&   € ÷ Tñ T˜sð T¨Cð T°Dñ Tr   c                ó   € R V 3RV33 FA  w  r#VR8  g   VR8”  d   \        V RV 24      hV^,          ^ 8w  g   K3  \        V RV 24      h	  \        W,          W,          4      pV^8”  d   \        RV  RV 24      hR# )	rZ   r[   é   i   z must be in [256, 2048], got z must be divisible by 16, got z&aspect ratio must be at most 6:1, got ÚxN)rG   Úmax)rZ   r[   r   ÚvalueÚratios   &&   r   Úvalidate_dimensionsrc   S   sš   € Ø  %Ð(¨8°VÐ*<Ó=‰ˆØ�3Œ;˜% $œ,Ü ˜vÐ%BÀ5À'ÐJÓKÐKØ�2�:˜Ž?Ü ˜vÐ%CÀEÀ7ÐKÓLÐLñ	 >ô
 �• ¥Ó/€EØˆq„yÜÐAÀ%ÀÈÈ&ÈÐRÓSÐSñ r   )N)Ú__conditional_annotations__Ú
__future__r   rS   Údataclassesr   Úpathlibr   Útypingr   r
   r   rB   r<   r   rJ   rX   rc   )rd   s   @r   Ú<module>ri      s”   øðß "Ð "ã Ý !Ý Ý ñ �$˜dÔ#÷2ð 2ó $ð2ñ �$˜dÔ#÷
ð 
ó $ð
ðB 6Ð ð Ñ,ØØ&ôð)€Ð
%ó ÷	Nõ/÷Tr   