From fc71e900a0528985263d770078181fa5cd6854e9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?C=C3=A9dric=20Deltheil?= Date: Sat, 21 Oct 2023 11:30:31 +0200 Subject: [PATCH] black --- .../foundationals/latent_diffusion/schedulers/ddpm.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/refiners/foundationals/latent_diffusion/schedulers/ddpm.py b/src/refiners/foundationals/latent_diffusion/schedulers/ddpm.py index 528d395..db004fa 100644 --- a/src/refiners/foundationals/latent_diffusion/schedulers/ddpm.py +++ b/src/refiners/foundationals/latent_diffusion/schedulers/ddpm.py @@ -56,9 +56,7 @@ class DDPM(Scheduler): else tensor(1, device=self.device) ) current_factor = current_cumulative_factor / previous_cumulative_scale_factor - estimated_denoised_data = ( - x - (1 - current_cumulative_factor) ** 0.5 * noise - ) / current_cumulative_factor**0.5 + estimated_denoised_data = (x - (1 - current_cumulative_factor) ** 0.5 * noise) / current_cumulative_factor**0.5 estimated_denoised_data = estimated_denoised_data.clamp(-1, 1) original_data_coeff = (previous_cumulative_scale_factor**0.5 * (1 - current_factor)) / ( 1 - current_cumulative_factor