
    3j                     v    S SK r S SKJr  S SKJrJr   " S S\R                  5      r " S S\5      rS\4S	 jr	g)
    N)override)ComfyExtensionioc                   T    \ rS rSr\S 5       r\SS\R                  4S jj5       rSr	g)ReferenceLatent   c                     [         R                  " SSSS[         R                  R                  S5      [         R                  R                  SSS9/[         R                  R                  5       /S	9$ )
Nr   zSet Reference Latentzmodel/conditioningzThis node sets the guiding latent for an edit model. If the model supports it you can chain multiple to set multiple reference images.conditioninglatentT)optional)node_iddisplay_namecategorydescriptioninputsoutputs)r   SchemaConditioningInputLatentOutput)clss    =/home/wildlama/comfy/ComfyUI/comfy_extras/nodes_edit_model.pydefine_schemaReferenceLatent.define_schema   sd    yy%/) a%%n5		48
 &&(
 	
    Nreturnc                 l    Ub  [         R                  " USUS   /0SS9n[        R                  " U5      $ )Nreference_latentssamplesT)append)node_helpersconditioning_set_valuesr   
NodeOutput)r   r
   r   s      r   executeReferenceLatent.execute   sC    '??ObekluevdwNx  BF  GL}}\**r    N)
__name__
__module____qualname____firstlineno__classmethodr   r   r$   r%   __static_attributes__r'   r   r   r   r      s3    
 
 +2== + +r   r   c                   L    \ rS rSr\S\\\R                        4S j5       r	Sr
g)EditModelExtension   r   c                    #    [         /$ 7fr(   )r   )selfs    r   get_node_list EditModelExtension.get_node_list   s      
 	
s   
r'   N)r)   r*   r+   r,   r   listtyper   	ComfyNoder4   r.   r'   r   r   r0   r0      s)    
T$r||*<%= 
 
r   r0   r   c                      [        5       $ r(   )r0   r'   r   r   comfy_entrypointr:   &   s    r   )
r"   typing_extensionsr   comfy_api.latestr   r   r8   r   r0   r:   r'   r   r   <module>r=      s7     & /+bll +0
 
 ,  r   