
    +j                     x    d dl Z d dlmZ d dlmZmZ  G d dej                  Z G d de          Zdefd	Z	dS )
    N)override)ComfyExtensionioc                   L    e Zd Zed             Zeddej        fd            ZdS )ReferenceLatentc           
          t          j        dddt           j                            d          t           j                            dd          gt           j                                        g          S )	Nr   z!advanced/conditioning/edit_modelszThis 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categorydescriptioninputsoutputs)r   SchemaConditioningInputLatentOutput)clss    =/home/wildlama/comfy/ComfyUI/comfy_extras/nodes_edit_model.pydefine_schemazReferenceLatent.define_schema   sl    y%8 a%%n55	488
 &&((
 
 
 	
    Nreturnc                 n    | t          j        |d|d         gid          }t          j        |          S )Nreference_latentssamplesT)append)node_helpersconditioning_set_valuesr   
NodeOutput)r   r	   r
   s      r   executezReferenceLatent.execute   sM    '?ObekluevdwNx  BF  G  G  GL}\***r   N)__name__
__module____qualname__classmethodr   r   r!   r"    r   r   r   r      s[        
 
 [
 + +2= + + + [+ + +r   r   c                   L    e Zd Zedeeej                          fd            ZdS )EditModelExtensionr   c                    K   t           gS r#   )r   )selfs    r   get_node_listz EditModelExtension.get_node_list   s       
 	
r   N)	r$   r%   r&   r   listtyper   	ComfyNoder-   r(   r   r   r*   r*      sE        
T$r|*<%= 
 
 
 X
 
 
r   r*   r   c                      t                      S r#   )r*   r(   r   r   comfy_entrypointr2   %   s    r   )
r   typing_extensionsr   comfy_api.latestr   r   r0   r   r*   r2   r(   r   r   <module>r5      s        & & & & & & / / / / / / / /+ + + + +bl + + +.
 
 
 
 
 
 
 
 ,            r   