make

Author: hlky

src/diffusers/pipelines/flux/pipeline_flux.py

@@ -15,6 +15,7 @@
 import inspect
 from typing import Any, Callable, Dict, List, Optional, Union
 
+import numpy as np
 import torch
 from transformers import (
     CLIPImageProcessor,

src/diffusers/schedulers/scheduling_euler_ancestral_discrete.py

@@ -20,7 +20,7 @@
 from ..configuration_utils import ConfigMixin, register_to_config
 from ..utils import BaseOutput, logging
 from ..utils.torch_utils import randn_tensor
-from .scheduling_utils import KarrasDiffusionSchedulers, SchedulerMixin, SamplingMixin
+from .scheduling_utils import KarrasDiffusionSchedulers, SamplingMixin, SchedulerMixin
 
 
 logger = logging.get_logger(__name__)  # pylint: disable=invalid-name

src/diffusers/schedulers/scheduling_euler_discrete.py

@@ -20,7 +20,7 @@
 from ..configuration_utils import ConfigMixin, register_to_config
 from ..utils import BaseOutput, logging
 from ..utils.torch_utils import randn_tensor
-from .scheduling_utils import KarrasDiffusionSchedulers, SchedulerMixin, SamplingMixin
+from .scheduling_utils import KarrasDiffusionSchedulers, SamplingMixin, SchedulerMixin
 
 
 logger = logging.get_logger(__name__)  # pylint: disable=invalid-name

src/diffusers/schedulers/scheduling_heun_discrete.py

@@ -20,7 +20,7 @@
 
 from ..configuration_utils import ConfigMixin, register_to_config
 from ..utils import BaseOutput
-from .scheduling_utils import KarrasDiffusionSchedulers, SchedulerMixin, SamplingMixin
+from .scheduling_utils import KarrasDiffusionSchedulers, SamplingMixin, SchedulerMixin
 
 
 @dataclass

src/diffusers/schedulers/scheduling_utils.py

@@ -247,6 +247,7 @@ def set_sigma_schedule(self, sigma_schedule: Union[Dict]):
         else:
             self._sigma_schedule = sigma_schedule
 
+
 class SamplingMixin:
     _step_index = None
     _begin_index = None
@@ -375,6 +376,5 @@ def get_velocity(self, sample: torch.Tensor, noise: torch.Tensor, timesteps: tor
         velocity = sqrt_alpha_prod * noise - sqrt_one_minus_alpha_prod * sample
         return velocity
 
-
     def __len__(self):
         return self.config.num_train_timesteps