diff --git a/src/diffusers/schedulers/scheduling_lms_discrete.py b/src/diffusers/schedulers/scheduling_lms_discrete.py index 0e1ed2049b..55dd3dbec8 100644 --- a/src/diffusers/schedulers/scheduling_lms_discrete.py +++ b/src/diffusers/schedulers/scheduling_lms_discrete.py @@ -130,14 +130,5 @@ class LMSDiscreteScheduler(SchedulerMixin, ConfigMixin): noisy_samples = (alpha_prod**0.5) * original_samples + ((1 - alpha_prod) ** 0.5) * noise return noisy_samples - def add_noise(self, original_samples, noise, timesteps): - sqrt_alpha_prod = self.alphas_cumprod[timesteps] ** 0.5 - sqrt_alpha_prod = self.match_shape(sqrt_alpha_prod, original_samples) - sqrt_one_minus_alpha_prod = (1 - self.alphas_cumprod[timesteps]) ** 0.5 - sqrt_one_minus_alpha_prod = self.match_shape(sqrt_one_minus_alpha_prod, original_samples) - - noisy_samples = sqrt_alpha_prod * original_samples + sqrt_one_minus_alpha_prod * noise - return noisy_samples - def __len__(self): return self.config.num_train_timesteps diff --git a/src/diffusers/utils/__init__.py b/src/diffusers/utils/__init__.py index ffa654c2e7..aef3827df1 100644 --- a/src/diffusers/utils/__init__.py +++ b/src/diffusers/utils/__init__.py @@ -28,12 +28,12 @@ from .import_utils import ( DummyObject, is_flax_available, is_inflect_available, + is_modelcards_available, is_scipy_available, is_tf_available, is_torch_available, is_transformers_available, is_unidecode_available, - is_modelcards_available, requires_backends, ) from .logging import get_logger