Update vllm_omni/core/sched/diffusion_scheduler.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Signed-off-by: Zhou Taichang <tzhouam@connect.ust.hk>
This commit is contained in:
Zhou TaichangandCopilot authored and GitHub committed 2025-10-22 17:44:20 +08:00
1 parent 8d6a360f96
commit b207bf89f2
1 file changed
+3 -1
+3 -1
View File
@@ -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