vllm_omni.diffusion.sched.request_scheduler ¶

RequestScheduler ¶

Scheduler for static request waves, including admission coalescing.

get_admission_wait_decision(
    *, now: float, dp_concurrent: bool = False
) -> _AdmissionWaitDecision

should_end_admission_wait(
    decision: _AdmissionWaitDecision,
    *,
    now: float,
    stable_since: float,
) -> bool

update_from_output(
    sched_output: DiffusionSchedulerOutput,
    output: BaseRunnerOutput,
) -> set[str]

build_request_batch_sampling_params_key(
    request: OmniDiffusionRequest,
) -> RequestBatchSamplingParamsKey

Build the compatibility key shared by scheduling and DP dispatch.