e39f4dddfb
Signed-off-by: Huang, Zeyu <11222265+fhfuih@users.noreply.github.com> Signed-off-by: Zeyu Huang | 黃澤宇 <11222265+fhfuih@users.noreply.github.com> Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com> Co-authored-by: Cursor <cursoragent@cursor.com>
33 lines
913 B
Python
33 lines
913 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
from vllm_omni.diffusion.sched.base_scheduler import BaseScheduler, SchedulerInterface
|
|
from vllm_omni.diffusion.sched.interface import (
|
|
CachedRequestData,
|
|
DiffusionRequestStatus,
|
|
DiffusionSchedulerOutput,
|
|
KVPrefetchJob,
|
|
NewRequestData,
|
|
SchedulerRequestState,
|
|
StepBatchSamplingParamsKey,
|
|
)
|
|
from vllm_omni.diffusion.sched.request_scheduler import RequestScheduler
|
|
from vllm_omni.diffusion.sched.step_scheduler import StepScheduler
|
|
|
|
Scheduler = RequestScheduler
|
|
|
|
__all__ = [
|
|
"DiffusionRequestStatus",
|
|
"CachedRequestData",
|
|
"DiffusionSchedulerOutput",
|
|
"KVPrefetchJob",
|
|
"NewRequestData",
|
|
"SchedulerRequestState",
|
|
"BaseScheduler",
|
|
"SchedulerInterface",
|
|
"StepBatchSamplingParamsKey",
|
|
"RequestScheduler",
|
|
"StepScheduler",
|
|
"Scheduler",
|
|
]
|