(beta_start, beta_end, num_diffusion_timesteps, warmup_frac)
| 13 | |
| 14 | |
| 15 | def _warmup_beta(beta_start, beta_end, num_diffusion_timesteps, warmup_frac): |
| 16 | betas = beta_end * np.ones(num_diffusion_timesteps, dtype=np.float64) |
| 17 | warmup_time = int(num_diffusion_timesteps * warmup_frac) |
| 18 | betas[:warmup_time] = np.linspace(beta_start, beta_end, warmup_time, dtype=np.float64) |
| 19 | return betas |
| 20 | |
| 21 | |
| 22 | def get_beta_schedule(beta_schedule, *, beta_start, beta_end, num_diffusion_timesteps): |