
    +j	                         S SK Jr  S SKJrJr   " S S\R
                  5      r " S S\R
                  5      r " S S\5      rS	\4S
 jr	g)    )override)ComfyExtensionioc                   l    \ rS rSr\S\R                  4S j5       r\S\R                  4S j5       r	Sr
g)CLIPTextEncodeControlnet   returnc                     [         R                  " SS[         R                  R                  S5      [         R                  R                  S5      [         R
                  R                  SSSS9/[         R                  R                  5       /SS9$ )	Nr   experimental/conditioningclipconditioningtextT)	multilinedynamic_promptsnode_idcategoryinputsoutputsis_experimental)r   SchemaClipInputConditioningStringOutputclss    7/home/wildlama/comfy/ComfyUI/comfy_extras/nodes_cond.pydefine_schema&CLIPTextEncodeControlnet.define_schema   sn    yy.0f%%%n5		$M
 __++-. 

 
	
    c                     UR                  U5      nUR                  USS9u  pV/ nU H:  nUS   US   R                  5       /n	XYS   S'   XiS   S'   UR                  U	5        M<     [        R
                  " U5      $ )NT)return_pooledr      cross_attn_controlnetpooled_output_controlnet)tokenizeencode_from_tokenscopyappendr   
NodeOutput)
r   r   r   r   tokenscondpooledctns
             r   execute CLIPTextEncodeControlnet.execute   s    t$..vT.JA1qtyy{#A,0aD()/5aD+,HHQK	 
 }}Qr"    N__name__
__module____qualname____firstlineno__classmethodr   r   r    r,   r3   __static_attributes__r5   r"   r   r   r      s>    
bii 
 
 	 "-- 	  	 r"   r   c                   l    \ rS rSr\S\R                  4S j5       r\S\R                  4S j5       r	Sr
g)T5TokenizerOptions!   r	   c                 ,   [         R                  " SS[         R                  R                  S5      [         R                  R                  SSSSSSS	9[         R                  R                  S
SSSSSS	9/[         R                  R                  5       /SS9$ )Nr>   r   r   min_paddingr   i'  r%   T)defaultminmaxstepadvanced
min_lengthr   )r   r   r   r   Intr   r   s    r   r     T5TokenizerOptions.define_schema"   s~    yy(0f%]A1%aZ^_\1!QY]^
 WW^^%& 

 
	
r"   c                     UR                  5       nS HE  nUR                  SR                  U5      U5        UR                  SR                  U5      U5        MG     [        R                  " U5      $ )N)t5xxl	pile_t5xlt5basemt5xlumt5xxlz{}_min_paddingz{}_min_length)cloneset_tokenizer_optionformatr   r,   )r   r   rA   rG   t5_types        r   r3   T5TokenizerOptions.execute0   s`    zz|KG%%&6&=&=g&FT%%o&<&<W&EzR L }}T""r"   r5   Nr6   r5   r"   r   r>   r>   !   s>    
bii 
 
 #r}} # #r"   r>   c                   L    \ rS rSr\S\\\R                        4S j5       r	Sr
g)CondExtension:   r	   c                 "   #    [         [        /$ 7fN)r   r>   )selfs    r   get_node_listCondExtension.get_node_list;   s      %
 	
s   r5   N)r7   r8   r9   r:   r   listtyper   	ComfyNoder[   r<   r5   r"   r   rV   rV   :   s)    
T$r||*<%= 
 
r"   rV   r	   c                     #    [        5       $ 7frY   )rV   r5   r"   r   comfy_entrypointra   C   s     ?s   N)
typing_extensionsr   comfy_api.latestr   r   r_   r   r>   rV   ra   r5   r"   r   <module>rd      sC    & / r||  6# #2
N 
 r"   