Schedulers

Scheduler

Abstract base class for schedulers of flow matching models.

OptimalTransportScheduler

Optimal transport scheduler which is commonly used to train flow matching models.

DiffusionScheduler

Scheduler for discrete-time diffusion models based on a given noise schedule.

NoiseSchedule

Abstract base class for noise schedules \(\sigma(t)\).

MemorylessNoiseSchedule

Memoryless noise schedule (https://arxiv.org/abs/2409.08861) based on the scheduler's eta function.

ConstantNoiseSchedule

Constant noise schedule with fixed sigma.