(beta_start, beta_end, num_diffusion_timesteps, warmup_frac)
| 56 | |
| 57 | |
| 58 | def _warmup_beta(beta_start, beta_end, num_diffusion_timesteps, warmup_frac): |
| 59 | betas = beta_end * np.ones(num_diffusion_timesteps, dtype=np.float64) |
| 60 | warmup_time = int(num_diffusion_timesteps * warmup_frac) |
| 61 | betas[:warmup_time] = np.linspace(beta_start, beta_end, warmup_time, dtype=np.float64) |
| 62 | return betas |
| 63 | |
| 64 | |
| 65 | def get_beta_schedule(beta_schedule, *, beta_start, beta_end, num_diffusion_timesteps): |