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:
1 parent
8d6a360f96
commit
b207bf89f2
1 file changed
+3
-1
@@ -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
|
||||
|
||||
Reference in new issue
Block a user