Skip to content

Commit 21ceda3

Browse files
committed
Remove duplicate add_noise
1 parent 5321f3e commit 21ceda3

File tree

2 files changed

+1
-10
lines changed

2 files changed

+1
-10
lines changed

src/diffusers/schedulers/scheduling_lms_discrete.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -130,14 +130,5 @@ def add_noise(self, original_samples, noise, timesteps):
130130
noisy_samples = (alpha_prod**0.5) * original_samples + ((1 - alpha_prod) ** 0.5) * noise
131131
return noisy_samples
132132

133-
def add_noise(self, original_samples, noise, timesteps):
134-
sqrt_alpha_prod = self.alphas_cumprod[timesteps] ** 0.5
135-
sqrt_alpha_prod = self.match_shape(sqrt_alpha_prod, original_samples)
136-
sqrt_one_minus_alpha_prod = (1 - self.alphas_cumprod[timesteps]) ** 0.5
137-
sqrt_one_minus_alpha_prod = self.match_shape(sqrt_one_minus_alpha_prod, original_samples)
138-
139-
noisy_samples = sqrt_alpha_prod * original_samples + sqrt_one_minus_alpha_prod * noise
140-
return noisy_samples
141-
142133
def __len__(self):
143134
return self.config.num_train_timesteps

src/diffusers/utils/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,12 +28,12 @@
2828
DummyObject,
2929
is_flax_available,
3030
is_inflect_available,
31+
is_modelcards_available,
3132
is_scipy_available,
3233
is_tf_available,
3334
is_torch_available,
3435
is_transformers_available,
3536
is_unidecode_available,
36-
is_modelcards_available,
3737
requires_backends,
3838
)
3939
from .logging import get_logger

0 commit comments

Comments
 (0)