From 999e429697b43fcbc9bd39611221e1a66e69128f Mon Sep 17 00:00:00 2001 From: Pierre Chapuis Date: Thu, 18 Jan 2024 14:27:30 +0100 Subject: [PATCH] fix bug in dpm_solver_first_order_update --- .../foundationals/latent_diffusion/schedulers/dpm_solver.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/refiners/foundationals/latent_diffusion/schedulers/dpm_solver.py b/src/refiners/foundationals/latent_diffusion/schedulers/dpm_solver.py index 3999a41..c2c4e16 100644 --- a/src/refiners/foundationals/latent_diffusion/schedulers/dpm_solver.py +++ b/src/refiners/foundationals/latent_diffusion/schedulers/dpm_solver.py @@ -51,7 +51,7 @@ class DPMSolver(Scheduler): def dpm_solver_first_order_update(self, x: Tensor, noise: Tensor, step: int) -> Tensor: current_timestep = self.timesteps[step] - previous_timestep = self.timesteps[step + 1 if step < len(self.timesteps) - 1 else 0] + previous_timestep = self.timesteps[step + 1] if step < self.num_inference_steps - 1 else tensor([0]) previous_ratio = self.signal_to_noise_ratios[previous_timestep] current_ratio = self.signal_to_noise_ratios[current_timestep]