vllm_omni.diffusion.models.schedulers.scheduling_dmd2_euler ¶

DMD2EulerScheduler ¶

Bases: FlowMatchEulerDiscreteScheduler

Euler scheduler that always uses the fixed DMD2 training timestep schedule.

set_timesteps ¶

set_timesteps(
    num_inference_steps: int | None = None,
    device: str | device | None = None,
    timesteps: list[int] | None = None,
    sigmas: list[float] | None = None,
    **kwargs,
) -> None