Diffusers.BetaSchedules.cosine_beta_scheduleFunction

Cosine beta schedule.

\[\overline{\alpha}_t = \cos \left( \frac{t / T + \epsilon}{1 + \epsilon} \frac{\pi}{2} \right)\]

Input

  • T::Int: number of timesteps
  • βₘₐₓ::Real=0.999f0: maximum value of β
  • ϵ::Real=1.0f-3: small value used to avoid division by zero

Output

  • β::Vector{Real}: βₜ values at each timestep t

References

source
Diffusers.BetaSchedules.exponential_beta_scheduleFunction

Exponential beta schedule.

\[\overline{\alpha}_t = \exp \left( \frac{-12 t}{T} \right)\]

Input

  • T::Int: number of timesteps
  • βₘₐₓ::Real=0.999f0: maximum value of β

Output

  • β::Vector{Real}: βₜ values at each timestep t
source