From 7eacd9ee1872f8d515dcdebdc0a831a6fc1227a1 Mon Sep 17 00:00:00 2001 From: AlpinDale Date: Wed, 27 Aug 2025 17:55:24 +0000 Subject: [PATCH] fix: priority scheduler crashing in V1 scheduler under high load --- aphrodite/v1/core/sched/scheduler.py | 1 + 1 file changed, 1 insertion(+) diff --git a/aphrodite/v1/core/sched/scheduler.py b/aphrodite/v1/core/sched/scheduler.py index 5ce64861db..ea4733356a 100644 --- a/aphrodite/v1/core/sched/scheduler.py +++ b/aphrodite/v1/core/sched/scheduler.py @@ -254,6 +254,7 @@ def schedule(self) -> SchedulerOutput: key=lambda r: (r.priority, r.arrival_time), ) self.running.remove(preempted_req) + scheduled_running_reqs.remove(preempted_req) else: preempted_req = self.running.pop()