o
    -iO                     @   s   d d Z dS )c           	      C   s   d|  d| d}| |krt |d ||vr5ddlm} || }||}||kr/t |d t |d || j}||  j}||krMt |d	| d
 d S )Nz&Specified KV sharing target layer for z is not valid: target layer  z(cannot be the same as the current layer.    )extract_layer_indexz#must come before the current layer.z,is not a valid Attention layer in the model.z,must be the same type as the current layer (z).)
ValueError vllm.model_executor.models.utilsr   	attn_type)	current_layer_nametarget_layer_namestatic_forward_context	error_msgr   current_layer_idxtarget_layer_idxtarget_layer_attn_typeexpected r   b/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/vllm/attention/utils/kv_sharing_utils.pyvalidate_kv_sharing_target   s(   

r   N)r   r   r   r   r   <module>   s   