diff --git a/src/refiners/foundationals/latent_diffusion/schedulers/ddim.py b/src/refiners/foundationals/latent_diffusion/schedulers/ddim.py index 200640d..18067bd 100644 --- a/src/refiners/foundationals/latent_diffusion/schedulers/ddim.py +++ b/src/refiners/foundationals/latent_diffusion/schedulers/ddim.py @@ -54,6 +54,4 @@ class DDIM(Scheduler): predicted_x = (x - sqrt(1 - current_scale_factor**2) * noise) / current_scale_factor denoised_x = previous_scale_factor * predicted_x + sqrt(1 - previous_scale_factor**2) * noise - self.previous_scale_factor = previous_scale_factor - return denoised_x