register trained_betas and timestep_values

2022-06-13 16:39:50 +02:00 · 2022-06-13 16:39:50 +02:00 · 61dc11c713
parent 29d9f02f83
commit 61dc11c713
1 changed files with 2 additions and 0 deletions
--- a/src/diffusers/schedulers/scheduling_ddpm.py
+++ b/src/diffusers/schedulers/scheduling_ddpm.py
@ -38,6 +38,8 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin):
            beta_start=beta_start,
            beta_end=beta_end,
            beta_schedule=beta_schedule,
+            trained_betas=trained_betas,
+            timestep_values=timestep_values,
            variance_type=variance_type,
            clip_predicted_image=clip_predicted_image,
        )