Skip to content

vllm_omni.diffusion.sched.step_scheduler ¶

logger `module-attribute` ¶

logger = init_logger(__name__)

StepScheduler ¶

Bases: BaseScheduler

Scheduler that advances each request by one denoise step per update.

add_request ¶

add_request(request: OmniDiffusionRequest) -> str

update_from_output ¶

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