From 97f92c6b4741e99ac278f326cabef4f7884d0e49 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nicol=C3=B2=20Lucchesi?= Date: Fri, 3 Apr 2026 17:40:16 +0200 Subject: [PATCH] [KVConnector] Skip `register_kv_caches` on profiling (#38558) Signed-off-by: NickLucche --- vllm/v1/worker/gpu_model_runner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/v1/worker/gpu_model_runner.py b/vllm/v1/worker/gpu_model_runner.py index bba707df0..98b3e56ab 100644 --- a/vllm/v1/worker/gpu_model_runner.py +++ b/vllm/v1/worker/gpu_model_runner.py @@ -6844,7 +6844,7 @@ class GPUModelRunner( # group self.drafter.validate_same_kv_cache_group(kv_cache_config) - if has_kv_transfer_group(): + if has_kv_transfer_group() and not is_profiling: kv_transfer_group = get_kv_transfer_group() if self.cross_layers_kv_cache is not None: assert self.cross_layers_attn_backend is not None