From df64f624c044e18071f178787b67e50f47c57028 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Tue, 14 Jun 2022 10:39:21 +0000 Subject: [PATCH] finish pndm --- src/diffusers/schedulers/scheduling_pndm.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/diffusers/schedulers/scheduling_pndm.py b/src/diffusers/schedulers/scheduling_pndm.py index fa1c9ca56d..cc27b52055 100644 --- 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]