[SchedulingPNDM ] reset cur_model_output after each call (#2376)

reset cur_model_output

[SchedulingPNDM ] reset cur_model_output after each call (#2376)
reset cur_model_output
de9ce9e9 · Suraj Patil · GitHub · fa35750d · de9ce9e9
Unverified Commit de9ce9e9 authored Feb 16, 2023 by Suraj Patil Committed by GitHub Feb 16, 2023
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
@@ -187,6 +187,7 @@ class PNDMScheduler(SchedulerMixin, ConfigMixin):

        self.ets = []
        self.counter = 0
+        self.cur_model_output = 0

    def step(
        self,