
    .`i                     .   d dl mZ d dlmZ d dlmZ d dlZd dlmZ  edej	        e
ej	                 z            Zeej	        gej	        f         Zeej	        gej	        f         Zeegef         Z ed	           G d
 d                      Zg dZdS )    )Callable)	dataclass)TypeVarN)PoolingParams_T)boundT)frozenc                   8    e Zd ZU dZeed<   	 d	dZdeddfdZdS )
PoolingParamsUpdateFrequires_token_idsotherreturnc                 :    t          | j        p|j                  S )Nr   )r   r   )selfr   s     |/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/model_executor/layers/pooler/common.py__or__zPoolingParamsUpdate.__or__   s'    "#6R%:R
 
 
 	
    paramsNc                     | j         |_         d S )Nr   )r   r   s     r   applyzPoolingParamsUpdate.apply   s    $($;!!!r   )r   r   r   r   )	__name__
__module____qualname__r   bool__annotations__r   r   r    r   r   r   r      s`         $$$$I
 
 
 

<M <d < < < < < <r   r   )ActivationFnClassifierFnProjectorFnr   )collections.abcr   dataclassesr   typingr   torchvllm.pooling_paramsr   Tensorlistr   r    r   r   r   __all__r   r   r   <module>r)      s   % $ $ $ $ $ ! ! ! ! ! !        - - - - - -WTU\(::;;;~u|3445b! $
< 
< 
< 
< 
< 
< 
< 
< Q
P
Pr   