black

finegrain-ai · Oct 21, 2023 · 87609bd · 87609bd
1 parent 98eded0
commit 87609bd
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/refiners/foundationals/latent_diffusion/schedulers/ddpm.py b/src/refiners/foundationals/latent_diffusion/schedulers/ddpm.py
@@ -56,9 +56,7 @@ def __call__(self, x: Tensor, noise: Tensor, step: int, generator: Generator | N
             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