reassign last noise uncond after model inference

v0xie · Jun 17, 2024 · 7393993 · 7393993
1 parent 08ab80e
commit 7393993
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/modules/sd_samplers_timesteps_impl.py b/modules/sd_samplers_timesteps_impl.py
@@ -176,6 +176,7 @@ def get_x_prev_and_pred_x0(e_t, noise_uncond, index):
             # Pseudo Improved Euler (2nd order)
             x_prev, pred_x0 = get_x_prev_and_pred_x0(e_t, last_noise_uncond, index)
             e_t_next = model(x_prev, t_next, **extra_args)
+            last_noise_uncond = model.last_noise_uncond
             e_t_prime = (e_t + e_t_next) / 2
         elif len(old_eps) == 1:
             # 2nd order Pseudo Linear Multistep (Adams-Bashforth)