diff --git a/third_party/vllm/vllm/v1/core/sched/scheduler.py b/third_party/vllm/vllm/v1/core/sched/scheduler.py index 5350c90..f817deb 100644 --- a/third_party/vllm/vllm/v1/core/sched/scheduler.py +++ b/third_party/vllm/vllm/v1/core/sched/scheduler.py @@ -2105,9 +2105,12 @@ class Scheduler(SchedulerInterface): req = self.requests[req_id] if req.status == RequestStatus.WAITING_FOR_REMOTE_KVS: self.finished_recving_kv_req_ids.add(req_id) - else: - assert RequestStatus.is_finished(req.status) + elif RequestStatus.is_finished(req.status): self._free_blocks(self.requests[req_id]) + else: + logger.debug( + "finished_recving for %s in status %s (partial remote prefill?)", + req_id, req.status) for req_id in kv_connector_output.finished_sending or (): logger.debug("Finished sending KV transfer for request %s", req_id) if req_id not in self.requests: