dmme.equations.iddpm#

dmme.equations.iddpm.cosine_schedule(timesteps: int = 4000, offset: float = 0.008)[source]#

Cosine variance schedule

Parameters:
  • timesteps – total timesteps

  • offset – small offset to prevent \(\beta_t\) from being too small near \(t = 0\)