From c7469ebe74a90bb8b704f3d5c1bdfe2d9240dabe Mon Sep 17 00:00:00 2001 From: Uranus Date: Wed, 28 Jun 2023 21:44:29 +0800 Subject: [PATCH] fix sde add noise typo (#3839) * fix sde typo * fix code style --- src/diffusers/schedulers/scheduling_sde_ve.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/diffusers/schedulers/scheduling_sde_ve.py b/src/diffusers/schedulers/scheduling_sde_ve.py index b92db0f048..339edfbb02 100644 --- a/src/diffusers/schedulers/scheduling_sde_ve.py +++ b/src/diffusers/schedulers/scheduling_sde_ve.py @@ -276,7 +276,11 @@ class ScoreSdeVeScheduler(SchedulerMixin, ConfigMixin): # Make sure sigmas and timesteps have the same device and dtype as original_samples timesteps = timesteps.to(original_samples.device) sigmas = self.discrete_sigmas.to(original_samples.device)[timesteps] - noise = torch.randn_like(original_samples) * sigmas[:, None, None, None] + noise = ( + noise * sigmas[:, None, None, None] + if noise is not None + else torch.randn_like(original_samples) * sigmas[:, None, None, None] + ) noisy_samples = noise + original_samples return noisy_samples