
    -`iO                         d  Z dS )c                 H   d|  d| d}| |k    rt          |dz             ||vrFddlm}  ||           } ||          }||k    rt          |dz             t          |dz             ||         j        }||          j        }||k    rt          |d	| d
z             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expecteds	            y/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/attention/utils/kv_sharing_utils.pyvalidate_kv_sharing_targetr      s   	;1C 	; 	;&7	; 	; 	; 
 ...%OOPPP 666HHHHHH
 0/0BCC../@AA 000Y)NNOOOY)WWXXX 44EFP%&89CH))SxSSSS
 
 	
 *)    N)r    r   r   <module>r      s   
 
 
 
 
r   