o
    iUz                     @   s  d dl Z G dd dZG dd deZG dd deZG dd	 d	eZG d
d deZG dd deZG dd deZG dd deZG dd deZ	G dd deZ
G dd deZG dd deZG dd deZG dd deZG dd deZG d d! d!eZG d"d# d#eZG d$d% d%eZG d&d' d'eZG d(d) d)eZG d*d+ d+eZG d,d- d-eZG d.d/ d/eZG d0d1 d1eZG d2d3 d3eZG d4d5 d5eZG d6d7 d7eZG d8d9 d9eZdS ):    Nc                   @   s<   e Zd ZdZdZdZdZdZdZdZ	dZ
dd Zdd	 ZdS )
LatentFormat      ?      N   c                 C   s
   || j  S Nscale_factorselflatent r   2/mnt/c/Users/fbmor/ComfyUI/comfy/latent_formats.py
process_in      
zLatentFormat.process_inc                 C   s
   || j  S r   r   r
   r   r   r   process_out   r   zLatentFormat.process_out)__name__
__module____qualname__r	   latent_channelslatent_dimensionslatent_rgb_factorslatent_rgb_factors_biaslatent_rgb_factors_reshapetaesd_decoder_namespacial_downscale_ratior   r   r   r   r   r   r      s    r   c                   @   s   e Zd ZdddZdS )SD15{P?c                 C   s.   || _ g dg dg dg dg| _d| _d S )N)gz?gAf?g,Ԛ?)g?gAf?gGz?)g%uҿgǺ?g`vOj?)gA`"˿g(пgڊetaesd_decoder)r	   r   r   )r   r	   r   r   r   __init__   s   
zSD15.__init__N)r   r   r   r   r   r   r   r   r   r      s    r   c                   @      e Zd ZdZdd ZdS )SDXLgy&1?c                 C   s2   g dg dg dg dg| _ g d| _d| _d S )N)g Ac]?ghs?gc]K?)gK46пJ4qgz6>W?)gq?g镲q?:pΈ)gK7AԿgϿgı.n̿)g6?Q둿g/nRtaesdxl_decoder)r   r   r   r   r   r   r   r   "   s   

zSDXL.__init__N)r   r   r   r	   r   r   r   r   r   r"      s    r"   c                   @   s$   e Zd Zdd Zdd Zdd ZdS )SDXL_Playground_2_5c                 C   sf   d| _ tg ddddd| _tg ddddd| _g dg dg dg d	g| _d
| _d S )Ng      ?)g?ܵgV-?gT㥛 g9v@   r   )gxz,C @gQڛ@g\m2@gfj@)gJ+?gs?g[Ӽ?)gqпngm4?)ZӼ?gj+?S)g	gӿg-1οg&Sʿr&   )r	   torchtensorviewlatents_meanlatents_stdr   r   r'   r   r   r   r   /   s   
zSDXL_Playground_2_5.__init__c                 C   6   | j |j|j}| j|j|j}|| | j | S r   r0   todevicedtyper1   r	   r   r   r0   r1   r   r   r   r   =      zSDXL_Playground_2_5.process_inc                 C   6   | j |j|j}| j|j|j}|| | j | S r   r3   r7   r   r   r   r   B   r8   zSDXL_Playground_2_5.process_outN)r   r   r   r   r   r   r   r   r   r   r(   .   s    r(   c                   @      e Zd Zdd ZdS )SD_X4c                 C   (   d| _ g dg dg dg dg| _d S )NghU?)gʡEͿgw#ؿg*DԿ)58EGr?g-臨?gz6>)g6<R!ҿgJ4οg[ A׿)g1%?I+vgH.!ſr	   r   r'   r   r   r   r   I      
zSD_X4.__init__Nr    r   r   r   r   r;   H       r;   c                   @   s   e Zd ZdZdZdd ZdS )SC_Prior   *   c                 C   sp   d| _ g dg dg dg dg dg dg dg d	g d
g dg dg dg dg dg dg dg| _d S )Nr   ){򰠿爅ZӼ㔿
F%u)g8d`Ŀg?ܥg_vO?)gEJY?g|a2U?gMb`)g:pΈҮea?g{?){Gz?ǺgH.!ƿ)#~jg|a2Ug[<Ŀ)g?W[?h|?5?goT?)l?gr?_vO?)tVƿg0*ȿg)g??g:#J{/?g?)g	h"l?gx$(~?gR!u?)gAf?㥛 ?gHPsׂ)gǺ?gp=
ףg_vOƿ)gb48ſgng()6?g-?g(\?)g/ngAfſglxz,Cпr?   r'   r   r   r   r   U   s$   
zSC_Prior.__init__N)r   r   r   r   r   r   r   r   r   r   rB   R   s    rB   c                   @   r!   )SC_Br   c                 C   r<   )Ng)kʚ@)g镲?gB?g|a2U0?)g'WʿgAǘg+)gӿgSÿg46<?)V-?g1%Ŀga+eڿr?   r'   r   r   r   r   l   r@   zSC_B.__init__N)r   r   r   r   r   r   r   r   r   rR   j   s    rR   c                   @   (   e Zd ZdZdd Zdd Zdd ZdS )	SD3rC   c                 C      d| _ d| _g dg dg dg dg dg dg d	g d
g dg dg dg dg dg dg dg dg| _g d| _d| _d S )Ngrh|?H.?)g|Pkr%   "lxz,?)gU؟?g8m4?g	h"l?)ggDio?V/?gHPs?)g'?z6>W[?,eX?)g(?g_Qڛ?ga+e)ga2U0*Cglxz,C???)g(~k	?gK=U?g?ܥ?)r#   g&S?gea?)gj+ݓMb`?g6[ ?)gHPg9v?S!uq)g|Pk?gDJ?gc=yX?)gPkw<,Ԫ?6)gSt$?+eX?gx?)lxz,C|~jthrK   )g2%䃮+ΧgX )g:pΈ¿g}гY¿g ~:p)gT?gS?gp=
ף?taesd3_decoderr	   shift_factorr   r   r   r'   r   r   r   r   w   *   

zSD3.__init__c                 C      || j  | j S r   rg   r	   r
   r   r   r   r         zSD3.process_inc                 C      || j  | j S r   r	   rg   r
   r   r   r   r      rk   zSD3.process_outNr   r   r   r   r   r   r   r   r   r   r   rU   u   
    rU   c                   @      e Zd ZdZdZdS )StableAudio1@   r)   Nr   r   r   r   r   r   r   r   r   rq          rq   c                   @   rT   )	FluxrC   c                 C   rV   )Ngxz,C?g=U?)geXgHP?gh o?)_Qk?g/$?g6<R?)g)\(?go_gfj+)5;NёgMb?gSt$?)gZڊ?gL
F%u?g*Dؠ?)g-C6:?g@߾?gZd;O?)g#~j?g{/L
?gCl?)g0*glgݓ)gJ+皙?rL   )g o_ι?gS?g<R!)g|?5^0*?/nR?)'?ga2U0*SH}m)gLJ?g/$?rX   )gvq-g}гYg[<)y&1gǺgx$(~)ge`TR'gkw#g	g곿g*DؠQ|agӼɵtaef1_decoderrf   r'   r   r   r   r      rh   zFlux.__init__c                 C   ri   r   rj   r
   r   r   r   r      rk   zFlux.process_inc                 C   rl   r   rm   r
   r   r   r   r      rk   zFlux.process_outNrn   r   r   r   r   ru      ro   ru   c                   @   ,   e Zd ZdZdZdd Zdd Zdd Zd	S )
Flux2   rC   c                 C   s  g g dg dg dg dg dg dg dg dg d	g d
g dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg d | _ g d!| _d"d# | _d S )$N)gw?gMSt$?g<,Ԛ}?)gMbX?lV}?gAf?)gׁsFga2U0*?g?ܵ|?)gS!uq?gsA?go_?)St$?gqog_vOv)2U0*?0*?gMb)g8d`r[   gMb)_Lu~jt?46<R?)g&Sֿgz6>W[ѿg&Sſ)g:pΈҞ給y):g;On)g'酿gq?g_LU?)gHP?;On?g)q?X5;N?gy&1|?)_QkMbpg+eX)-C6j?:pΈ?gMb?)*	g-C6JΈ?)H}]?g"uq?_L?)gZӼ?gǺv?rG   )ga2U0*cg/nbrc   )g.1?g_Leg9vz?)g-C6?gMO?gZӼ?)g?߾?gH}}gǺF)r   gHPsׂ?g0*x)        g 	gnt)rd   rF   q)r   a2U0*S?gF%u{)rS   r   {Gzd?)gI+?r   a2U0*s)gSt$gڊeggj+) rhg	g?rW   )ŏ1w-!ga2U0*3?g"uq)gAǘgZd;Ogor~   c              	   S   s\   |  | jd ddd| jd | jd dddddd	 | jd d| jd d | jd d S )
Nr       r   r)   r         )reshapeshapepermute)tr   r   r   <lambda>   s   \ z Flux2.__init__.<locals>.<lambda>)r   r   r   r'   r   r   r   r      s   	
 
#zFlux2.__init__c                 C      |S r   r   r
   r   r   r   r         zFlux2.process_inc                 C   r   r   r   r
   r   r   r   r      r   zFlux2.process_outNr   r   r   r   r   r   r   r   r   r   r   r   r      s    'r   c                   @   r   )
Mochi   r   c                 C   s   d| _ tg dd| jddd| _tg dd| jddd| _g dg dg dg dg d	g d
g dg dg dg dg dg dg| _g d| _d | _	d S )Nr   )gX(;gRP:vg:*$gV*~g]̴%?ghgҩy?g>1g[bdg,,
g[aogkn|r)   )g?gı?gPo*T?g`4?gq@a?g7k1Z?gAtŽ?g\4?gT~`4?g	(V?gDcd?gˏ?)rb   g;Onrr   )g 	?geXg?߾)镲?gŏ1w-!_?g0*?)gng2U0*?g
h"lxz?)g(\gS!uq?gAc]Kȿ)笭_vO~?S㥛Af)g/ngec]ea)gD?g,Ԛ?g,Ԛ?)rE   gͪզgQI);pΈްK=U?K=U)gl?gJY8ǺV?)gׁsFgMbXg?Ɯ)gMbgQI&¿g)0¿)
r	   r-   r.   r/   r   r0   r1   r   r   r   r'   r   r   r   r      s,   

zMochi.__init__c                 C   r2   r   r3   r7   r   r   r   r   
  r8   zMochi.process_inc                 C   r9   r   r3   r7   r   r   r   r     r8   zMochi.process_outN)r   r   r   r   r   r   r   r   r   r   r   r   r      s    r   c                   @   s    e Zd ZdZdZdZdd ZdS )LTXVr   r   r   c                 C   s  g g dg dg dg dg dg dg dg dg d	g d
g dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg d g d!g d"g d#g d$g d%g d&g d'g d(g d)g d*g d+g d,g d-g d.g d/g d0g d1g d2g d3g d4g d5g d6g d7g d8g d9g d:g d;g d<g d=g d>g d?g d@g dAg dBg dCg dDg dEg dFg dGg dHg dIg dJg dKg dLg dMg dNg dOg dPg dQg dRg dSg dTg dUg dVg dWg dXg dYg dZg d[g d\g d]g d^g d_g d`g dag dbg dcg ddg deg dfg dgg dhg dig djg dkg dlg dmg dng dog dpg dqg drg dsg dtg dug dvg dwg dxg dyg dzg d{g d|g d}g d~g dg d| _ g d| _d S )N)g&2?g0Dgߋ/ㅄ)gr2q ?ggҦٰ?gUx|CO?)gI%rgIg%P7p)g7C?gCBJMagcKE`e?)g3I]n?g1|DL$?gt?)gTsP?g$ug6b(l?)ga824g8H-gxe6)gALgݱ&geF%`?)g5Xg">|g"1A)g7A\g?{x>?gfzk)gh[:?gLaHs?gR<k`)gó?g37:Ba?gĐ?)gД~g_3uyg7R?)g>&H~?gS ?g^<g=?)gyr?O?gvێ_.hgCO}:)gZ8?gF[DA?g[X7?)gBr?g.=_;?g9|)g1 =giSulj?gAr?)gRZ[?gu?gq5+-?)g2g*cadgEOli)g?RDU?g.|?gbG,?)gMpgC-Rgz[(Y?)g?x=gLg,Cʿ)gĭ?gW?g_um)g֜\sgIO6?g>?)go+6[gfsg:Ru)gӢ>6ag{%9`Wc?g'5?)gߣzgXЕgV县)gkA!g&mͅgfpzrz)g=sg(8t?gQZ{)gfh<y?gnʆ5?gH_?)gY:zg{`_g!)g,ص}?gbPWg%z)g2_wNg'էy?g2|?)gBg^-g|	)g>?gS!uqg͋)g
ug@'?g}i@s?)g܀#ÿg+yp˿gL;˿)gxρ?gbf(|?gG$)gha?gː^dg|ڥ)gT6ہo?g<&xgPn)g_(qg_{fIgdZZ)gPE4pvgubT4g<h)gq,!zg|<i?g-VЄ)gQJtgBc
I}g\>/hcM)g9ܝgʃ[cg= }Q?)g!Lnog1oÃg!Y?)g%Gt?gѱJ\ǈ?g}r?)gw'-?g)%?giȀ?)gp'vg3gOVZp)gvӂggGq::)g-4h?guݑ?geU*?)gYZgsdg.=4=r)gB?g.5#?g)?)g@JڎgWlBg^fh?)g#k$	g
)g(H0̊?)glg?gdMz)goA[sEi?gVCK??g3IbI?)g0[wTg^gpB!)gviagH¾Dg\qqTn)gUg$B#gayFQ?gZUP?)g"T3k)gXiTgqIx)gGǁgA^Tg;^y?)gTTJëgډH?gѺWy?)gcU6br?g|
~gobHN&n)gкorgGYZ?gG=D;?)ge ?gaz3?gaɰp?)gLƄgġm]r?gמY?)g?+zwgNG[hS?gj?)geRsng)ugѭT?)g/M.?g`eM,?gx'?)gR'᡿gxE옿gC p왿)g9yg4g))gɐcg;J"g,)gdDXCagfg#/yh)g`}~S?gEԽ]qg|)gp9M?gMde{g )gn2n?gXİØgES)g\Y"T?g-VЄ?gHLP?)gm?gb<1?g%#gaO;?)ghhsgϫKmgpF?)gr)s?gZ0xugɾud?)g7[Ԃr6xg~yg.y])g"]?gI*ݍ?g%K_?)gĻ}~?gBZ?goH'ۀ)gR>G?gZ?g @p)gNK[wg:>f^fg)p)g{\?gGC?gj	i?)g3 Kg=b?g/\yr?)gR҇?gl.HD?g4})g"?gFe?g:6U?)g82Fu?gj;n?g UܸŌ?)g5?ҒgzؘgJ)g
THC?gt[wg׻?)gPng[R,g=zÍ)g
p?gP9?gLl	TZz?)gUk?gN6g(֐)gWj2>'XgA{gYe \?)g[%XΜ?gmtOq?g>4aɠ?)gǷwg1LgC)gh?~g3%gA
)g#SogaTR'?gLn?)g۲~g Uܸgx[y)g]Yl?tl?gE>'Ig})gۂ?gPp?gKEx g?)gcW}?geK&XgYI/o)gy):b?g	\?gEn)g Q?gy7?gI?)gwF[Dvg.:Yjߘgd6:)g:
k~?gg?RD?g';?)g)(Ry?gU*y?gb?)g~dŐ?g*?(9q?gkn{?)gb0̕?gŐL*?gŭڗ?)gՍg@h g)g{c ?!vy?g_.{?)go{gkjP?g<ύ?)g;߫y?g>?ml?go>9me)gK'Ⱦfg8
KTxgÁ傿)gL2r?g.?Na~?gpzr:)g]o%g._x%ɓ?g,Z2N)gc?g JAsg|x #)giߚ?g1!撚?gKXc'?)gw\?gf*#?gsf?)gagv)Vs?g<L)g?2ʝ~g)g`~?r   )gq#D/goXVg1?)gcϞ?g-Ƀ?go)gY9鍿gtt{gq89@p?)g46<g=yX5ſg2U0*пr   r   r'   r   r   r   r     s
  	
 !"#$%&'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\]^_`abcdefghijklmnopqrstuvwxyz{|}~   zLTXV.__init__N)r   r   r   r   r   r   r   r   r   r   r   r     s
    r   c                   @   r:   )LTXAVc                 C   s   d | _ d | _d S r   r   r'   r   r   r   r     s   
zLTXAV.__init__Nr    r   r   r   r   r     rA   r   c                   @   s   e Zd ZdZdZdZg dg dg dg dg dg d	g d
g dg dg dg dg dg dg dg dg dgZg dZdZdS )HunyuanVideorC   r   g>I?)gMbX9gm{gbX9Ȧ?)g5;Nѱ?g'1Z?gݓ?)gS㥋?gx&1g!u)g_vO?gg/$)r   r   gǺ?)g0*g
F%ug6[)g^g6?g6[ )aTR'g^K=g{Gz)g5;NѱgX9vg,C⶿)gjMStg_Lg333333)g}bٽ?gޓZ?g-?)gmҽ?rN   gfa֤?)goʡͿg~jtȿgMbX)r   眢#?gh o)g 	ggDiogPkw)gV}b?gc]Fg\ Acſ)gݓ?ga2U0*g9#J{taehvN)	r   r   r   r   r   r	   r   r   r   r   r   r   r   r     s.    r   c                   @   s   e Zd ZdZdZg dg dg dg dg dg dg d	g d
g dg dg dg dg dg dg dg dgZg dZdS )Cosmos1CV8x8x8rC   r   )gsA?g46<?g鷯?)X gJ4g1%ӿ)ga+e޿gZd;ۿg rhٿ)giq?g6[ ?g`TR'?)gˡEgzGế{Gz?)gn?gO@a?gJY8?)gClǿgHP˿r   )z):˯g<Nё\˿r   )g^)?g\m?g9v?)g??go_?g{Gz?)gH}?g&?g&)g	cZ¿gHP׿g~jt׿)gK7A?g<,?g/$?)g$C׿g6[ ɿgDioԿ)gZd;ǿrP   go_?),C?r=   gǺ?)g;MOgvq-ȿgDJɿN)r   r   r   r   r   r   r   r   r   r   r   r     s*    r   c                   @   s   e Zd ZdZdZg dg dg dg dg dg dg d	g d
g dg dg dg dg dg dg dg dgZg dZdd Zdd Zdd Z	dS )Wan21rC   r   )g-g=yXſg?)gŏ1w-?gDioɤ?rH   )gͪ?gTN?g;Nё\?)gΈ?gX ?gV-?)Ǻ?gݓZ?g	^)ˠ)笭_vO?gb=yg/$)gm4@ֿgё\CٿgQҿ)r   gc=yX?gHzG?)g0*gz6>W?gR!u)g鷯rM   gD?)g rh?gd`TR?g!u?)r   g:H?g6[?)gc]FrY   g:H?)g~jtx?g8m4Mb@?)gm4@?gQ?gzG?)ga+e?go_?g?)grh|ǿgJY8g/$տc                 C   sP   d| _ tg dd| jddd| _tg dd| jddd| _d| _d S )Nr   )gy):gMOg^)gQ?rO   gZӼ?gBfjÿgU0*?gL
F%u?gMg&?gz6>׿gF%uȿg[ AcgMJ?gW2ıҿr)   )g_L@gNё\C?gQ@g?@g9#J{?g|a2U?gHPs@g0* @gJ{/L&
@gJY8@g]C@g(?gK46?gS:?go_Ι@g-?lighttaew2_1)r	   r-   r.   r/   r   r0   r1   r   r'   r   r   r   r     s   
zWan21.__init__c                 C   r2   r   r3   r7   r   r   r   r     r8   zWan21.process_inc                 C   r9   r   r3   r7   r   r   r   r     r8   zWan21.process_outN)
r   r   r   r   r   r   r   r   r   r   r   r   r   r   r     s0    r   c                   @   s  e Zd ZdZdZdZg g dg dg dg dg dg d	g d
g dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg d g d!g d"g d#g d$g d%g d&g d'g d(g d)g d*g d+g d,g d-g d.g d/g d0g d1g d2g d3Zg d4Zd5d6 Zd7S )8Wan220   r   rC   )tF_?r   gHPsr?)gV/g o_ΩgL7A`?)gy&1?D?gkw#?)gU0*дgioTg?)g	^)˰?gӼɵ?gMO?)F%u?g3?g+eX?)h|?5?{?gh|?5?)gHPr   r   )r   爅ZӼㄿH}?)rd   g;OnˡE)gQ|a?g0*?	^)ˀ?)gj+ݳg?ܵ?g<,Ժ?)g!u?r   gw#?)a2U0*C?gBi?gAf?)g|a2Ur`   gTN)r^   ga2U0*c?9v?)g@ǘ?gA`"?d`T?)gQ|ar   gOec)gv/g{Pkg&䃞ͪ)gJ{/L?gt$~?gUد?)g"ug&gZӼt)g+?gx&?g46<R?)glxz,C|?H}?r   )r   go_gvq-?)r   Oecg)gPkw¿g,eX-)g48EG?g@a+?r   )gŏ1w-!r   r*   )r   ga2U0*?gZӼ?)gk	ggBigǺV)r+   g_)Ǻ?g oŏ?)r>   gX9vg~jt)b=y?gZd;gp=
ף?)g~k	g&S:y&1)g_LgUN@sHPsb)r   gǺ?Έ)^K=rz   gm{?)g_v?g?g?)gHPsעgl?r   )g/nֿg[B>٬ֿgݵ|гɿ)JY(~k	g/$)gL7A`尿r   rI   )gMO_Q{?r   )gDlgMSt$gV/)gV-	^)p?g~jtX?)g<R!?g,eX?gHP?)g o_Ω?g_)Ǻr   )g٬\mJ4?gI+v?)~:?gzgW2c                 C   sP   d| _ d| _tg dd| jddd| _tg dd| jddd| _d S )Nr   lighttaew2_2)0g_LͿr   ga4g8gDioͿ'ѿ5;Nё?gI&?g\C?gTR'rQ   gX ?gsA϶?gjt?gng333333?g\(\ǿg>W[̿g.1澿gBiޱgEJY?guV?g$~ʿg=U˿gc]F?gMʿ?ܵ|g[ Ac?g?߾?gg鷯?r   grh|rJ   g!uqſggj+?gn4@?gfj+ҿgʿg9vgx#?gŏ1w-!?g镲?g&W?gHPsҿgY ?g?߾g:MſgN@r)   )0g"~?g0*?gZӼ?g1?gqh ?gV-?gͪ?g{/L
?g%u?g鷯?gec]?g镲q?gx&1?g|?5^?g:M?g$?gY8m?g_L?gE?gKY8?g7d?g3?goT?gQ?gW[?gZӼ?g9#J?gd]Fx?g{/L
?g<Nё\?g-?g-1?g0*D?g rh?gI+?g??g46<R?g}8gD?gŏ1w?gMJ?g`vOj?g(\?g46<?gGr?g$(~k?gfj+?g?g@?)r	   r   r-   r.   r/   r   r0   r1   r'   r   r   r   r   B  s   zWan22.__init__N)	r   r   r   r   r   r   r   r   r   r   r   r   r   r     s    	
 !"#$%&'()*+,-./03r   c                   @   s(  e Zd ZdZdZdZdZg g dg dg dg dg d	g d
g dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg d g d!g d"g d#g d$g d%g d&g d'g d(g d)g d*g d+g d,g d-g d.g d/g d0g d1g d2g d3g d4g d5g d6g d7g d8g d9g d:g d;g d<g d=g d>g d?g d@g dAg dBg dCg dDZg dEZdFS )GHunyuanImage21rr   r   r   gĬ?)g 	gjMS[B>٬)r   gΈ?r   )gS㥛gɳr   )gDJa4憎gV}b)ggsr   g}гY)gd`Tg|гYgB`"۩)gvq-?X9v?r   )r   g{Pk?ry   )Q?gHP?gH}m?)g,C⦿g(r   )gŏ1w-!or   g㥛 )g=yXr   g()gqo?gHPsע?r   )gS㥛İ?JY8?gHP?)gJ+?gڊe?g48EG?)g$۷?gb48?gp=
ף?)gΈr   S㥋)gMbr   g_vO)gZӼg0*rz   )g8gDio%ug1%䳿)gy&1lgŏ1w-!?r   )g&S?glV}?g0'?)r   r   g/n?)r   g46<R?r   )g镲g+e⨿g&S)r   r   g	c?)rx   r   gR!u?)r   g~jtrw   )gMSt$?g_L?g+eX?)gx$(~?gOe?g#?)r   ?߾H}])gǺr\   r   )gW2ġg<,Ԛ?r   )gj+?gd`T?ry   )r   r   g
ףp=
?)g=yX?gHzG?r_   )a+eig_vOf?g&1)r   g"~j?gL7A`?)gŏ1w-!r   gu?)r,   r   gK7A)gJY8ֵ?g<,?g 	?)_Qڛr   gH})gI+?r   r   )gx&?r   r|   )g"ur   g(0)rG   gT㥛 gX9v)gǘg;On?g&?)g<Nё\gh ogD)g镢?gmV}b?gx$(~?)gtF_?rv   r   )r   g{Pk?r   )go?g>yX5ͻ?gGx$(?)r{   r   gˡE?)r   geX?r   )gH}gKg&S)g"u?r   g2U0*)r   DJg&)ga4?g o_Ή?g-C6z?)I+g_LUr   )gDioɔgUN@r   )g+	hgI.!?gea)g[B>٬?g58EGr?gy):?)r   gj+r}   )g~jtxr   ggj+)gǺF?g-C6g0*N)	r   r   r   r   r   r   r	   r   r   r   r   r   r   r   V  s   	
 !"#$%&'()*+,-./0123456789:;<=>?@Br   c                   @   s(   e Zd ZdZdZdZdd Zdd ZdS )	HunyuanImage21Refinerrr   r   fk}Ж?c                 C   s   || j  }tj|d d d d d df |fdd}|ddddd}|j\}}}}}|||d d| ||}|ddddd }|S )Nr)   r   )dimr   r   r   )r	   r-   catr   r   r   
contiguous)r   r   outb	f_times_2chwr   r   r   r     s   
(z HunyuanImage21Refiner.process_inc                 C   s   || j  }|ddddd}|j\}}}}}|||d|d ||}|dddddd||d |d ||}|ddddd}|d d d d dd f }|S )Nr   r   r)   r   r   r   )r	   r   r   r   )r   r   zr   fr   r   r  r   r   r   r     s   
*z!HunyuanImage21Refiner.process_outN)r   r   r   r   r   r	   r   r   r   r   r   r   r     s    	r   c                   @   s,  e Zd Zg g dg dg dg dg dg dg dg dg d	g d
g dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg d Zg d!Zd"Zd#Zd$Zd%Zd&Z	d'S )(HunyuanVideo15)r+   r   g<,Ԋ)r   g7A`в?r   )r   gV/?glxz,C)r   gZӼt?r   )gI+?gPkw?rN   )g	^)pr   gEԨ)gA`"?g0L
F%?cZB>?)gB>٬?ݵ|г?rZ   )gec]?gΈr   )g1w-!?g0*?g2w-!?)r]   r  g_vOn?)g_Q[r   r   )ggj+?g46<R?gDioɔ?)gr?gz?g$ۗ)r   g(0?geX?)gjM?gg?r   )g}гY?gQg~jth?)g/nr   gB>٬)r   g@ǈg镢)gSt$gUدr   )gQ|a2?gJY?ghs?)gV-?gK=U?r   )g%u?r   r   )gw-!l?gŏ1w-!o?rQ   )gJ4?g-!lVgF%uk)gǘr   gDioɤ)r   gh|?5g1w-!)r   g46<Rr  )r   gcZB?gPs?)r   r   gK=U)r   r$   gHPs?)r   ggsg@ǘ?)ra   gMOg?ܵ|r   r   rC   r   lighttaehy1_5N)
r   r   r   r   r   r   r   r   r	   r   r   r   r   r   r    s    	
 #r  c                   @      e Zd ZdZdZdZdS )Hunyuan3Dv2rr   r)   g֣M?Nr   r   r   r   r   r	   r   r   r   r   r	        r	  c                   @   r  )Hunyuan3Dv2_1g@_.?rr   r)   N)r   r   r   r	   r   r   r   r   r   r   r    r  r  c                   @   r  )Hunyuan3Dv2minirr   r)   gy M?Nr
  r   r   r   r   r    r  r  c                   @   rp   )ACEAudior   r   Nrs   r   r   r   r   r    rt   r  c                   @   rp   )
ACEAudio15rr   r)   Nrs   r   r   r   r   r    rt   r  c                   @   r   )
ChromaRadiancer   r)   c                 C   s   g dg dg dg| _ d S )N)r   r   r   )r   r   r   )r   r   r   )r   r'   r   r   r   r     s   
zChromaRadiance.__init__c                 C   r   r   r   r
   r   r   r   r     r   zChromaRadiance.process_inc                 C   r   r   r   r
   r   r   r   r   	  r   zChromaRadiance.process_outNr   r   r   r   r   r    s    r  c                   @   s   e Zd ZdZdS )ZImagePixelSpaceu   Pixel-space latent format for ZImage DCT variant.
    No VAE encoding/decoding — the model operates directly on RGB pixels.
    N)r   r   r   __doc__r   r   r   r   r    s    r  )r-   r   r   r"   r(   r;   rB   rR   rU   rq   ru   r   r   r   r   r   r   r   r   r   r   r  r	  r  r  r  r  r  r  r   r   r   r   <module>   s<    
  1* 1NJ+