Skip to content

Commit c7469eb

Browse files
authored
fix sde add noise typo (huggingface#3839)
* fix sde typo * fix code style
1 parent 1500130 commit c7469eb

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/diffusers/schedulers/scheduling_sde_ve.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -276,7 +276,11 @@ def add_noise(
276276
# Make sure sigmas and timesteps have the same device and dtype as original_samples
277277
timesteps = timesteps.to(original_samples.device)
278278
sigmas = self.discrete_sigmas.to(original_samples.device)[timesteps]
279-
noise = torch.randn_like(original_samples) * sigmas[:, None, None, None]
279+
noise = (
280+
noise * sigmas[:, None, None, None]
281+
if noise is not None
282+
else torch.randn_like(original_samples) * sigmas[:, None, None, None]
283+
)
280284
noisy_samples = noise + original_samples
281285
return noisy_samples
282286

0 commit comments

Comments
 (0)