Skip to content

vllm_omni.diffusion.sched.step_scheduler ¶

logger `module-attribute` ¶

logger = init_logger(__name__)

StepScheduler ¶

Bases: _BaseScheduler

Placeholder scheduler that advances a request one denoise step per update.

add_request ¶

add_request(request: OmniDiffusionRequest) -> str

schedule ¶

schedule() -> DiffusionSchedulerOutput

update_from_output ¶

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