diffusers/pr_14178/en/api/schedulers/helios_dmd.md

HeliosDMDScheduler

HeliosDMDScheduler is based on the pyramidal flow-matching sampling introduced in Helios.

HeliosDMDScheduler[[diffusers.HeliosDMDScheduler]]

initialize the global timesteps and sigmas

Init the timesteps for each stage

• begin_index (int) -- The begin index for the scheduler.

Sets the begin index for the scheduler. This function should be run from pipeline before the inference.

Setting the timesteps and sigmas for each stage

• mu (float) -- The mu parameter for the time shift.
• sigma (float) -- The sigma parameter for the time shift.
• t (torch.Tensor) -- The input timesteps.torch.TensorThe time-shifted timesteps.

Apply time shifting to the sigmas.

scheduling_helios_dmd