diff --git a/vllm/v1/core/sched/scheduler.py b/vllm/v1/core/sched/scheduler.py index 7fc4776b0261..a09d1c36169d 100644 --- a/vllm/v1/core/sched/scheduler.py +++ b/vllm/v1/core/sched/scheduler.py @@ -1287,5 +1287,7 @@ def _update_from_kv_xfer_finished(self, logger.debug("Finished recving KV transfer for request %s", req_id) self.finished_recving_kv_req_ids.add(req_id) for req_id in (kv_connector_output.finished_sending or ()): + if req_id not in self.requests: + continue logger.debug("Finished sending KV transfer for request %s", req_id) self._free_blocks(self.requests[req_id])