Schedulers
Abstract base class for schedulers of flow matching models. |
|
Optimal transport scheduler which is commonly used to train flow matching models. |
|
Scheduler for discrete-time diffusion models based on a given noise schedule. |
|
Abstract base class for noise schedules \(\sigma(t)\). |
|
Memoryless noise schedule (https://arxiv.org/abs/2409.08861) based on the scheduler's eta function. |
|
Constant noise schedule with fixed sigma. |