Wu JIAZHENandjader
3fd4a4dc27
Signed-off-by: jader <yjader@foxmail.com> Signed-off-by: asukaqaq-s <1311722138@qq.com> Co-authored-by: jader <yjader@foxmail.com>
28 lines
723 B
Python
28 lines
723 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
|
|
from vllm_omni.diffusion.sched.step_scheduler import StepScheduler
|
|
|
|
Scheduler = RequestScheduler
|
|
|
|
__all__ = [
|
|
"DiffusionRequestStatus",
|
|
"CachedRequestData",
|
|
"DiffusionRequestState",
|
|
"DiffusionSchedulerOutput",
|
|
"NewRequestData",
|
|
"SchedulerInterface",
|
|
"RequestScheduler",
|
|
"StepScheduler",
|
|
"Scheduler",
|
|
]
|