Commit 61dc11c7 authored by patil-suraj's avatar patil-suraj
Browse files

register trained_betas and timestep_values

parent 29d9f02f
......@@ -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,
)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment