61cd532cff
Signed-off-by: jader <yjader@foxmail.com> Signed-off-by: JiangJie Zhang <76905040+yJader@users.noreply.github.com> Co-authored-by: SYLAR <125541396+lishunyang12@users.noreply.github.com> Co-authored-by: Didan Deng <33117903+wtomin@users.noreply.github.com>
26 lines
635 B
Python
26 lines
635 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
from vllm_omni.diffusion.sched.interface import (
|
|
CachedRequestData,
|
|
DiffusionRequestState,
|
|
DiffusionRequestStatus,
|
|
DiffusionSchedulerOutput,
|
|
NewRequestData,
|
|
SchedulerInterface,
|
|
)
|
|
from vllm_omni.diffusion.sched.request_scheduler import RequestScheduler
|
|
|
|
Scheduler = RequestScheduler
|
|
|
|
__all__ = [
|
|
"CachedRequestData",
|
|
"DiffusionRequestState",
|
|
"DiffusionRequestStatus",
|
|
"DiffusionSchedulerOutput",
|
|
"NewRequestData",
|
|
"RequestScheduler",
|
|
"Scheduler",
|
|
"SchedulerInterface",
|
|
]
|