
    3j}                     ~    S SK Jr  S SKrS SKrS SKJrJr   " S S\R                  5      r	 " S S\5      r
S\
4S	 jrg)
    )overrideN)ComfyExtensionioc                   4    \ rS rSr\S 5       r\S 5       rSrg)SD_4XUpscale_Conditioning   c                    [         R                  " SS[         R                  R                  S5      [         R                  R                  S5      [         R                  R                  S5      [         R
                  R                  SSSS	S
S9[         R
                  R                  SSSSSSS9/[         R                  R                  SS9[         R                  R                  SS9[         R                  R                  SS9/S9$ )Nr   z,model/conditioning/stable diffusion upscalerimagespositivenegativescale_ratiog      @g        g      $@g{Gz?)defaultminmaxstepnoise_augmentation      ?gMbP?T)r   r   r   r   advanced)display_namelatent)node_idcategoryinputsoutputs)r   SchemaImageInputConditioningFloatOutputLatent)clss    </home/wildlama/comfy/ComfyUI/comfy_extras/nodes_sdupscale.pydefine_schema'SD_4XUpscale_Conditioning.define_schema   s    yy/Cx(%%j1%%j1}csSWX3ScsY^imn &&J&?&&J&?		  h 7
 	
    c                    [        S[        UR                  S   U-  5      5      n[        S[        UR                  S   U-  5      5      n[        R                  R                  UR                  SS5      S-  S-
  US-  US-  SS	5      n/ n	/ n
U H:  nUS
   US   R                  5       /nXS   S'   X\S   S'   U	R                  U5        M<     U H:  nUS
   US   R                  5       /nXS   S'   X\S   S'   U
R                  U5        M<     [        R                  " UR                  S
   SUS-  US-  /5      n[        R                  " XSU05      $ )N   g       @r      bilinearcenterr   concat_imager   samples)r   roundshapecomfyutilscommon_upscalemovedimcopyappendtorchzerosr   
NodeOutput)r"   r
   r   r   r   r   widthheightpixelsout_cpout_cntnr   s                 r#   execute!SD_4XUpscale_Conditioning.execute   sZ   AuV\\"-;<=Qfll2.<=>++V^^Bq-AC-G3,NPUYZPZ\bfg\gisu}~A1qtyy{#A#)aD );aD%&MM!	  A1qtyy{#A#)aD );aD%&MM!	  fll1oq&A+uzJK}}Vi-?@@r&    N)__name__
__module____qualname____firstlineno__classmethodr$   rC   __static_attributes__rE   r&   r#   r   r      s*    
 
$ A Ar&   r   c                   L    \ rS rSr\S\\\R                        4S j5       r	Sr
g)SdUpscaleExtension5   returnc                    #    [         /$ 7fN)r   )selfs    r#   get_node_list SdUpscaleExtension.get_node_list6   s      &
 	
s   
rE   N)rF   rG   rH   rI   r   listtyper   	ComfyNoderS   rK   rE   r&   r#   rM   rM   5   s)    
T$r||*<%= 
 
r&   rM   rO   c                     #    [        5       $ 7frQ   )rM   rE   r&   r#   comfy_entrypointrY   =   s     s   )typing_extensionsr   r9   comfy.utilsr3   comfy_api.latestr   r   rW   r   rM   rY   rE   r&   r#   <module>r]      s=    &   /+A +A\
 
  2  r&   