o
    
۾i                     @   s   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	 B d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d	G d
d dZg dZdS )    )Callable)	dataclass)TypeVarN)PoolingParams_T)boundT)frozenc                   @   s8   e Zd ZU dZeed< 	 dddZdeddfd	d
ZdS )PoolingParamsUpdateFrequires_token_idsotherreturnc                 C   s   t | jp|jdS )Nr
   )r	   r
   )selfr    r   \/home/ubuntu/.local/lib/python3.10/site-packages/vllm/model_executor/layers/pooler/common.py__or__   s   
zPoolingParamsUpdate.__or__paramsNc                 C   s   | j |_ d S )Nr   )r   r   r   r   r   apply   s   zPoolingParamsUpdate.apply)r   r	   r   r	   )	__name__
__module____qualname__r
   bool__annotations__r   r   r   r   r   r   r   r	      s
   
 
r	   )ActivationFnClassifierFnProjectorFnr	   )collections.abcr   dataclassesr   typingr   torchvllm.pooling_paramsr   Tensorlistr   r   r   r   r	   __all__r   r   r   r   <module>   s   