finish pndm

df64f624 · Patrick von Platen · f0a99e76 · df64f624
Commit df64f624 authored Jun 14, 2022 by Patrick von Platen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/diffusers/schedulers/scheduling_pndm.py src/diffusers/schedulers/scheduling_pndm.py +1 -0

No files found.
--- 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]