25 lines
635 B
Python
25 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",
|
||
|
|
]
|