From b207bf89f2c63da50aff0eebb318765edc6aa485 Mon Sep 17 00:00:00 2001 From: Zhou Taichang Date: Wed, 22 Oct 2025 17:44:20 +0800 Subject: [PATCH] Update vllm_omni/core/sched/diffusion_scheduler.py Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> Signed-off-by: Zhou Taichang --- vllm_omni/core/sched/diffusion_scheduler.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/vllm_omni/core/sched/diffusion_scheduler.py b/vllm_omni/core/sched/diffusion_scheduler.py index 30ba0c4e0..2c9e10c99 100644 --- a/vllm_omni/core/sched/diffusion_scheduler.py +++ b/vllm_omni/core/sched/diffusion_scheduler.py @@ -1,4 +1,6 @@ -from vllm.v1.core.sched.scheduler import SchedulerOutput, EngineCoreOutputs, Request, RequestStatus, SpecDecodingStats, defaultdict, Optional +from vllm.v1.core.sched.scheduler import SchedulerOutput, EngineCoreOutputs, Request, RequestStatus, SpecDecodingStats +from collections import defaultdict +from typing import Optional from vllm.v1.core.sched.request_queue import create_request_queue from vllm.v1.engine import EngineCoreEventType from vllm.distributed.kv_events import KVEventBatch