vllm-omni/vllm_omni/diffusion/sched/__init__.py

# 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",
]