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

YuechuanLi · Oct 20, 2022 · 4a76e5d · 4a76e5d
1 parent 83f8a5f
commit 4a76e5d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/diffusers/schedulers/scheduling_pndm.py b/src/diffusers/schedulers/scheduling_pndm.py
@@ -310,6 +310,7 @@ def step_plms(
         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