# HeliosDMDScheduler

`HeliosDMDScheduler` is based on the pyramidal flow-matching sampling introduced in [Helios](https://huggingface.co/papers).

## 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.Tensor`The time-shifted timesteps.

Apply time shifting to the sigmas.

scheduling_helios_dmd