Skip to content

GitLab

Unverified Commit da7d4cf2 authored by nicolas-dufour's avatar nicolas-dufour Committed by GitHub
Browse files

[BugFix]: Fixed add_noise in LMSDiscreteScheduler (#253) 



* Fixed add_noise in LMSDiscreteScheduler

* Linting

* Update src/diffusers/schedulers/scheduling_lms_discrete.py
Co-authored-by: default avatarAnton Lozhkov <aglozhkov@gmail.com>
Co-authored-by: default avatarAnton Lozhkov <aglozhkov@gmail.com>
parent 9e1b1ca4
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
src/diffusers/schedulers/scheduling_lms_discrete.py
View file @ da7d4cf2
...@@ -124,10 +124,8 @@ class LMSDiscreteScheduler(SchedulerMixin, ConfigMixin): ...@@ -124,10 +124,8 @@ class LMSDiscreteScheduler(SchedulerMixin, ConfigMixin):
return {"prev_sample": prev_sample} return {"prev_sample": prev_sample}
def add_noise(self, original_samples, noise, timesteps): def add_noise(self, original_samples, noise, timesteps):
alpha_prod = self.alphas_cumprod[timesteps] sigmas = self.match_shape(self.sigmas, noise)
alpha_prod = self.match_shape(alpha_prod, original_samples) noisy_samples = original_samples + noise * sigmas[timesteps]
noisy_samples = (alpha_prod**0.5) * original_samples + ((1 - alpha_prod) ** 0.5) * noise
return noisy_samples return noisy_samples
def __len__(self): def __len__(self):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment