File tree Expand file tree Collapse file tree 1 file changed +12
-5
lines changed Expand file tree Collapse file tree 1 file changed +12
-5
lines changed Original file line number Diff line number Diff line change @@ -58,12 +58,19 @@ def __init__(
58
58
"""
59
59
super ().__init__ ()
60
60
61
- if not envs .VLLM_USE_V1 :
62
- assert kv_sharing_target_layer_name is None , NotImplementedError (
63
- "KV sharing is not supported in V0." )
64
-
65
- # Verify target layer is valid for cross-layer KV sharing
66
61
if kv_sharing_target_layer_name is not None :
62
+ if not envs .VLLM_USE_V1 :
63
+ raise NotImplementedError (
64
+ "Cross-layer KV sharing is not supported in V0." )
65
+
66
+ if has_kv_transfer_group ():
67
+ # TODO(sarckk): look into gaps for making cross-layer
68
+ # KV sharing work with KV connectors
69
+ raise NotImplementedError (
70
+ "Cross-layer KV sharing is not yet compatible "
71
+ "with KV connector." )
72
+
73
+ # Verify target layer is valid for cross-layer KV sharing
67
74
from vllm .model_executor .models .utils import extract_layer_index
68
75
current_layer_idx = extract_layer_index (prefix )
69
76
target_layer_idx = extract_layer_index (
You can’t perform that action at this time.
0 commit comments