From ac0277361a2b0657c8c8774be94b70aa3782174c Mon Sep 17 00:00:00 2001
From: scxue <xueshuchen17@mails.ucas.edu.cn>
Date: Mon, 18 Dec 2023 21:19:48 +0800
Subject: [PATCH] arrange for alphabet order in init file

---
 src/diffusers/__init__.py                       | 4 ++--
 src/diffusers/schedulers/__init__.py            | 4 ++--
 src/diffusers/schedulers/scheduling_sasolver.py | 4 +++-
 3 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/src/diffusers/__init__.py b/src/diffusers/__init__.py
index a87a0bab6125..174338a95001 100644
--- a/src/diffusers/__init__.py
+++ b/src/diffusers/__init__.py
@@ -152,12 +152,12 @@
             "LCMScheduler",
             "PNDMScheduler",
             "RePaintScheduler",
+            "SASolverScheduler",
             "SchedulerMixin",
             "ScoreSdeVeScheduler",
             "UnCLIPScheduler",
             "UniPCMultistepScheduler",
             "VQDiffusionScheduler",
-            "SASolverScheduler",
         ]
     )
     _import_structure["training_utils"] = ["EMAModel"]
@@ -528,12 +528,12 @@
             LCMScheduler,
             PNDMScheduler,
             RePaintScheduler,
+            SASolverScheduler,
             SchedulerMixin,
             ScoreSdeVeScheduler,
             UnCLIPScheduler,
             UniPCMultistepScheduler,
             VQDiffusionScheduler,
-            SASolverScheduler,
         )
         from .training_utils import EMAModel
 
diff --git a/src/diffusers/schedulers/__init__.py b/src/diffusers/schedulers/__init__.py
index aae4e4afb9ab..a99c82cfdd99 100644
--- a/src/diffusers/schedulers/__init__.py
+++ b/src/diffusers/schedulers/__init__.py
@@ -60,12 +60,12 @@
     _import_structure["scheduling_lcm"] = ["LCMScheduler"]
     _import_structure["scheduling_pndm"] = ["PNDMScheduler"]
     _import_structure["scheduling_repaint"] = ["RePaintScheduler"]
+    _import_structure["scheduling_sasolver"] = ["SASolverScheduler"]
     _import_structure["scheduling_sde_ve"] = ["ScoreSdeVeScheduler"]
     _import_structure["scheduling_unclip"] = ["UnCLIPScheduler"]
     _import_structure["scheduling_unipc_multistep"] = ["UniPCMultistepScheduler"]
     _import_structure["scheduling_utils"] = ["KarrasDiffusionSchedulers", "SchedulerMixin"]
     _import_structure["scheduling_vq_diffusion"] = ["VQDiffusionScheduler"]
-    _import_structure["scheduling_sasolver"] = ["SASolverScheduler"]
 
 try:
     if not is_flax_available():
@@ -151,12 +151,12 @@
         from .scheduling_lcm import LCMScheduler
         from .scheduling_pndm import PNDMScheduler
         from .scheduling_repaint import RePaintScheduler
+        from .scheduling_sasolver import SASolverScheduler
         from .scheduling_sde_ve import ScoreSdeVeScheduler
         from .scheduling_unclip import UnCLIPScheduler
         from .scheduling_unipc_multistep import UniPCMultistepScheduler
         from .scheduling_utils import KarrasDiffusionSchedulers, SchedulerMixin
         from .scheduling_vq_diffusion import VQDiffusionScheduler
-        from .scheduling_sasolver import SASolverScheduler
 
     try:
         if not is_flax_available():
diff --git a/src/diffusers/schedulers/scheduling_sasolver.py b/src/diffusers/schedulers/scheduling_sasolver.py
index c5dc915b96cd..03a493e6a64b 100644
--- a/src/diffusers/schedulers/scheduling_sasolver.py
+++ b/src/diffusers/schedulers/scheduling_sasolver.py
@@ -16,9 +16,11 @@
 # The codebase is modified based on https://github.com/huggingface/diffusers/blob/main/src/diffusers/schedulers/scheduling_dpmsolver_multistep.py
 
 import math
+from typing import List, Optional, Tuple, Union, Callable
+
 import numpy as np
 import torch
-from typing import List, Optional, Tuple, Union, Callable
+
 from ..configuration_utils import ConfigMixin, register_to_config
 from ..utils.torch_utils import randn_tensor
 from .scheduling_utils import KarrasDiffusionSchedulers, SchedulerMixin, SchedulerOutput