o
ia @ s d Z ddlZddlmZ ddlZddlZddlmZm Z G dd de j
Z G dd de j
Z G d d
d
e j
Z
G d d d e j
ZG d
d deZdefddZdS )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 @ s& e Zd Zedd ZedddZdS ) StableCascade_EmptyLatentImagec
C s~ t jddt jjdddtjddt jjdddtjddt jjd d
d d d
ddt jjdd
d
ddgt jjddt jjddgdS )Nr latent/stable_cascadewidthi )defaultminmaxstepheight compression* Tr
r r r
advanced
batch_sizei )r
r r stage_c display_namestage_bnode_idcategoryinputsoutputs)r SchemaIntInputnodesMAX_RESOLUTIONLatentOutputcls r) ?/mnt/c/Users/fbmor/ComfyUI/comfy_extras/nodes_stable_cascade.py
define_schema s z,StableCascade_EmptyLatentImage.define_schemar c C sH t |d|| || g}t |d|d |d g}td|id|iS )N r samples)torchzerosr
NodeOutput)r( r r r r c_latentb_latentr) r) r* execute- s z&StableCascade_EmptyLatentImage.executeN)r __name__
__module__ __qualname__ classmethodr+ r3 r) r) r) r* r s
r c @ $ e Zd Zedd Zedd ZdS ) StableCascade_StageC_VAEEncodec
C sT t jddt jdt jdt jjddddd d
d gt jjd d
t jjdd
gdS )Nr: r imagevaer r r r r Tr r r r r )r r Imager" Vaer! r% r& r' r) r) r* r+ 9 s
z,StableCascade_StageC_VAEEncode.define_schemac C s |j d }|j d }|| |j }|| |j }tj|dd||dddd}||d d d d d d d df } t| j d d |d
d |d
d g}
t
d | id |
iS )
Nr bicubiccenter r r r r- ) shapedownscale_ratiocomfyutilscommon_upscalemovedimencoder. r/ r r0 ) r( r; r< r r r out_width
out_heightsr1 r2 r) r) r* r3 I s
$&(z&StableCascade_StageC_VAEEncode.executeNr4 r) r) r) r* r: 8 s
r: c @ r9 )!StableCascade_StageB_Conditioningc C s. t jddt jdt jdgt j gdS )NrP zconditioning/stable_cascade conditioningr r )r r Conditioningr" r% r& r' r) r) r* r+ \ s
z/StableCascade_StageB_Conditioning.define_schemac C sF g }|D ]}|d }|d |d< |d |g}|| qt|S )Nr r- stable_cascade_priorr )copyappendr r0 )r( rQ r ctdnr) r) r* r3 j s
z)StableCascade_StageB_Conditioning.executeNr4 r) r) r) r* rP [ s
rP c @ r9 )'StableCascade_SuperResolutionControlnetc
C sL t jdddt jdt jdgt jjddt jjddt jjd dgd
S ) NrZ z_for_testing/stable_cascadeTr; r< controlnet_inputr r r )r r is_experimentalr r )r r r= r" r> r& r% r' r) r) r* r+ v s
z5StableCascade_SuperResolutionControlnet.define_schemac C s |j d }|j d }|j d }||d d d d d d d df dd}t|d|d |d g}t|d|d |d g}t|d
|id
|iS ) Nr? r@ r rD r rA r, r rE r- )rF rL rK r. r/ r r0 ) r( r; r< r r r r[ r1 r2 r) r) r* r3 s
.z/StableCascade_SuperResolutionControlnet.executeNr4 r) r) r) r* rZ u s
rZ c @ s( e Zd Zedeeej fddZdS )StableCascadeExtensionreturnc s t tttgS N)r rP r: rZ )selfr) r) r*
get_node_list s z$StableCascadeExtension.get_node_listN) r5 r6 r7 r listtyper ComfyNodera r) r) r) r* r] s r] r^ c s t S r_ )r] r) r) r) r* comfy_entrypoint s re )__doc__r. typing_extensionsr comfy.utilsrH r# comfy_api.latestr r rd r r: rP rZ r] re r) r) r) r* s #"