From 4a0d6ac40bb8a35e2cf43c11754bc51b49c8a73d Mon Sep 17 00:00:00 2001 From: Robert Shaw Date: Wed, 24 Sep 2025 14:41:00 -0400 Subject: [PATCH] updated Signed-off-by: Robert Shaw --- vllm/v1/core/sched/scheduler.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/vllm/v1/core/sched/scheduler.py b/vllm/v1/core/sched/scheduler.py index 7fc4776b02611..2b2cd63c2e20f 100644 --- a/vllm/v1/core/sched/scheduler.py +++ b/vllm/v1/core/sched/scheduler.py @@ -1288,4 +1288,9 @@ class Scheduler(SchedulerInterface): self.finished_recving_kv_req_ids.add(req_id) for req_id in (kv_connector_output.finished_sending or ()): logger.debug("Finished sending KV transfer for request %s", req_id) - self._free_blocks(self.requests[req_id]) + if req_id not in self.requests: + logger.warning( + "Got finished sending KV transfer for request %s," + "but the request is already freed.", req_id) + else: + self._free_blocks(self.requests[req_id])