diffusion_models.gaussian_diffusion.beta_schedulers

Module Contents

class BaseBetaScheduler(steps, enforce_zero_terminal_snr=False)[source]
steps[source]
betas[source]
alpha_bars[source]
enforce_zero_terminal_snr()[source]
abstract sample_betas()[source]
abstract compute_alpha_bar()[source]
to(device)[source]
classmethod from_tensors(steps, betas, alpha_bars)[source]
class LinearBetaScheduler(beta_start=0.0001, beta_end=0.02, steps=1000, enforce_zero_terminal_snr=True)[source]
beta_start[source]
beta_end[source]
sample_betas()[source]
compute_alpha_bar()[source]
class CosineBetaScheduler(offset=0.008, steps=1000, max_beta=0.999)[source]
offset[source]
max_beta[source]
steps[source]
f(t)[source]
sample_betas()[source]
compute_alpha_bar()[source]