3j S r SSKrSSKJr SSKrSSKrSSKJrJ r " S S\ R 5 r " S S\ R 5 r " S S
\ R 5 r
" S S \ R 5 r " S
S\5 rS\4S jrg)a
This file is part of ComfyUI.
Copyright (C) 2024 Stability AI
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see .
N)override)ComfyExtensionioc 8 \ rS rSr\S 5 r\SS j5 rSrg) StableCascade_EmptyLatentImage c [ R " SS[ R R SSS[ R
SS9[ R R SSS[ R
SS9[ R R S S
S S S
SS9[ R R SS
S
SS9/[ R R SS9[ R R SS9/S9$ )Nr model/latent/stable cascadewidthi )defaultminmaxstepheight compression* Tr r r r advanced
batch_sizei )r r r stage_c display_namestage_bnode_idcategoryinputsoutputs)r SchemaIntInputnodesMAX_RESOLUTIONLatentOutputclss A/home/wildlama/comfy/ComfyUI/comfy_extras/nodes_stable_cascade.py
define_schema,StableCascade_EmptyLatentImage.define_schema s yy42 WdAUAU\] ^ Xt%BVBV]^ _ ]BA3QY] ^ \1! F i 8 i 8
c [ R " USX#- X- /5 n[ R " USUS- US- /5 n[ R " SU0SU05 $ )N r samples)torchzerosr
NodeOutput)r, r r r r c_latentb_latents r- execute&StableCascade_EmptyLatentImage.execute- se ;;
B0EuG[ \];;
Av{EQJ GH}} x
x
r0 N)r __name__
__module__ __qualname____firstlineno__ classmethodr. r9 __static_attributes__r; r0 r- r r s(
r0 r c 4 \ rS rSr\S 5 r\S 5 rSrg) StableCascade_StageC_VAEEncode8 c Z [ R " SS[ R R S5 [ R R S5 [ R
R SSSSS S
S 9/[ R R S S
9[ R R SS
9/S9$ )NrD r
imagevaer r r r r Tr r r r r )r r$ Imager& Vaer% r) r* r+ s r- r. ,StableCascade_StageC_VAEEncode.define_schema9 s yy42w' U# ]BA3QY] ^ i 8 i 8
r0 c UR S nUR S nXC- UR - nXS- UR - n[ R R UR SS5 XgSS5 R SS5 nUR
US S 2S S 2S S 2S S24 5 n [ R " U R S S US
- S - US
- S - /5 n
[ R " S U 0S U
05 $ )
Nr bicubiccenter r r r
r3 ) shapedownscale_ratiocomfyutilscommon_upscalemovedimencoder4 r5 r r6 ) r, rG rH r r r out_width
out_heightsr7 r8 s r- r9 &StableCascade_StageC_VAEEncode.executeI s BR )S-@-@@ +s/B/BB
KK & &u}}R':IS\^f g o opqrt u::a!Abqbk*;;q 11v{a6G%ST*XYIY Z[}} x
x
r0 r; Nr< r; r0 r- rD rD 8 s(
r0 rD c 4 \ rS rSr\S 5 r\S 5 rSrg)!StableCascade_StageB_Conditioning[ c [ R " SS[ R R S5 [ R R S5 /[ R R 5 /S9$ )Nr` z!model/conditioning/stable cascade conditioningr r )r r$ Conditioningr& r) r* r+ s r- r. /StableCascade_StageB_Conditioning.define_schema\ sW yy78%%n5 *
&&(
r0 c / nU H6 nUS R 5 nUS US' US U/nUR U5 M8 [ R " U5 $ )Nr r3 stable_cascade_priorr )copyappendr r6 )r, rc r ctdns r- r9 )StableCascade_StageB_Conditioning.executej sY A! A(/ (:A$ %1q A
HHQK
}}Q r0 r; Nr< r; r0 r- r` r` [ s(
r0 r` c 4 \ rS rSr\S 5 r\S 5 rSrg)'StableCascade_SuperResolutionControlnetu c R [ R " SSS[ R R S5 [ R R S5 /[ R R SS9[ R R SS9[ R R S S9/S
9$ ) Nrp zexperimental/stable_cascadeTrG rH controlnet_inputr r r )r r! is_experimentalr" r# )r r$ rI r& rJ r* r) r+ s r- r. 5StableCascade_SuperResolutionControlnet.define_schemav s yy=2 w' U#
-?@ i 8 i 8
r0 c t UR S nUR S nUR S nUR US S 2S S 2S S 2S S24 5 R SS5 n[ R " USUS- US- /5 n[ R " USUS - US - /5 n[
R " US
U0S
U05 $ ) NrM rN r rR r rO r2 r rS r3 )rT rZ rY r4 r5 r r6 ) r, rG rH r r r rs r7 r8 s r- r9 /StableCascade_SuperResolutionControlnet.execute s BR [[^
::eAa"1"Ho6>>q"E;;
B" erk JK;;
Av{EQJ GH}} - x0
x
r0 r; Nr< r; r0 r- rp rp u s(
r0 rp c L \ rS rSr\S\\\R 4S j5 r Sr
g)StableCascadeExtension returnc 6 # [ [ [ [ /$ 7fN)r r` rD rp )selfs r-
get_node_list$StableCascadeExtension.get_node_list s
+ - * 3
s r; N) r= r> r? r@ r listtyper ComfyNoder rB r; r0 r- ry ry s)
T$r||*<%=
r0 ry r{ c # [ 5 $ 7fr} )ry r; r0 r- comfy_entrypointr s ! ##s
)__doc__r4 typing_extensionsr comfy.utilsrV r' comfy_api.latestr r r r rD r` rp ry r r; r0 r- r sv $
& / R\\ : R\\ F 4 bll D
^
$ 6 $r0