finish pndm

2022-06-14 10:39:21 +00:00 · 2022-06-14 10:39:21 +00:00 · df64f624c0
parent f0a99e7684
commit df64f624c0
1 changed files with 1 additions and 0 deletions
--- a/src/diffusers/schedulers/scheduling_pndm.py
+++ b/src/diffusers/schedulers/scheduling_pndm.py
@ -96,6 +96,7 @@ class PNDMScheduler(SchedulerMixin, ConfigMixin):
        return self.time_steps[num_inference_steps]

    def step_warm_up(self, residual, image, t, num_inference_steps):
+        # TODO(Patrick) - need to rethink whether the "warmup" way is the correct API design here
        warmup_time_steps = self.get_warmup_time_steps(num_inference_steps)

        t_prev = warmup_time_steps[t // 4 * 4]