diffusion_model

Functions

find_noise_schedule()

Classes

CosineNoiseSchedule([min_log_snr, ...])	Cosine noise schedule for diffusion models.
DiffusionModel(*args, **kwargs)	Diffusion Model as described in this overview paper [1].
EDMNoiseSchedule([sigma_data, sigma_min, ...])	EDM noise schedule for diffusion models.
NoiseSchedule(name, variance_type[, weighting])	Noise schedule for diffusion models.