move checks out of unified_kv_cache_update custom op (#33943)

Signed-off-by: Rohan138 <rohanpotdar138@gmail.com>
2026-02-07 07:30:09 -06:00
parent ce9b3cd3e9
commit de3869bb4d
7 changed files with 79 additions and 100 deletions
--- a/vllm/model_executor/models/whisper_causal.py
+++ b/vllm/model_executor/models/whisper_causal.py
@@ -172,6 +172,9 @@ def create_whisper_attention_backend_with_block_pooling(
            if (
                not underlying_attn_backend.forward_includes_kv_cache_update
                and attn_metadata is not None
+                and layer.kv_sharing_target_layer_name is None
+                and key is not None
+                and value is not None
            ):
                self.do_kv_cache_update(
                    layer, key, value, kv_cache, attn_metadata.slot_mapping