[PNDM Scheduler] Make sure list cannot grow forever (#882)

4a76e5d4 · Patrick von Platen · GitHub · 83f8a5ff · 4a76e5d4
Unverified Commit 4a76e5d4 authored Oct 20, 2022 by Patrick von Platen Committed by GitHub Oct 20, 2022
Show 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
@@ -310,6 +310,7 @@ class PNDMScheduler(SchedulerMixin, ConfigMixin):
        prev_timestep = timestep - self.config.num_train_timesteps // self.num_inference_steps

        if self.counter != 1:
+            self.ets = self.ets[-3:]
            self.ets.append(model_output)
        else:
            prev_timestep = timestep