o
    i                     @   sX   d dl Z d dlmZ d dlmZmZ G dd dejZG dd deZdefd	d
Z	dS )    N)override)ComfyExtensionioc                   @   s.   e Zd Zedd ZeddejfddZdS )ReferenceLatentc              	   C   s4   t jdddt jdt jjdddgt j gd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)cls r   ;/mnt/c/Users/fbmor/ComfyUI/comfy_extras/nodes_edit_model.pydefine_schema   s   
zReferenceLatent.define_schemaNreturnc                 C   s,   |d urt j|d|d gidd}t|S )Nreference_latentssamplesT)append)node_helpersconditioning_set_valuesr   
NodeOutput)r   r   r   r   r   r   execute   s   
zReferenceLatent.executeN)__name__
__module____qualname__classmethodr   r   r   r   r   r   r   r   r      s
    
r   c                   @   s(   e Zd Zedeeej  fddZdS )EditModelExtensionr   c                    s   t gS r   )r   )selfr   r   r   get_node_list   s   z EditModelExtension.get_node_listN)	r    r!   r"   r   listtyper   	ComfyNoder&   r   r   r   r   r$      s    r$   r   c                   C   s   t  S r   )r$   r   r   r   r   comfy_entrypoint%   s   r*   )
r   typing_extensionsr   comfy_api.latestr   r   r)   r   r$   r*   r   r   r   r   <module>   s    