test_backend_registration

Author: hlky

src/diffusers/schedulers/__init__.py

@@ -74,7 +74,6 @@
     _import_structure["scheduling_unipc_multistep"] = ["UniPCMultistepScheduler"]
     _import_structure["scheduling_utils"] = ["AysSchedules", "KarrasDiffusionSchedulers", "SchedulerMixin"]
     _import_structure["scheduling_vq_diffusion"] = ["VQDiffusionScheduler"]
-    _import_structure["sigmas"] = ["BetaSigmas", "ExponentialSigmas", "KarrasSigmas"]
 
 try:
     if not is_flax_available():
@@ -175,7 +174,6 @@
         from .scheduling_unipc_multistep import UniPCMultistepScheduler
         from .scheduling_utils import AysSchedules, KarrasDiffusionSchedulers, SchedulerMixin
         from .scheduling_vq_diffusion import VQDiffusionScheduler
-        from .sigmas import BetaSigmas, ExponentialSigmas, KarrasSigmas
 
     try:
         if not is_flax_available():

src/diffusers/schedulers/sigmas/__init__.py

@@ -19,6 +19,7 @@
     OptionalDependencyNotAvailable,
     _LazyModule,
     get_objects_from_module,
+    is_scipy_available,
     is_torch_available,
     is_transformers_available,
 )
@@ -35,10 +36,21 @@
 
     _dummy_objects.update(get_objects_from_module(dummy_pt_objects))
 else:
-    _import_structure["beta_sigmas"] = ["BetaSigmas"]
     _import_structure["exponential_sigmas"] = ["ExponentialSigmas"]
     _import_structure["karras_sigmas"] = ["KarrasSigmas"]
 
+try:
+    if not (is_torch_available() and is_scipy_available()):
+        raise OptionalDependencyNotAvailable()
+except OptionalDependencyNotAvailable:
+    from ...utils import dummy_torch_and_scipy_objects  # noqa F403
+
+    _dummy_objects.update(get_objects_from_module(dummy_torch_and_scipy_objects))
+
+else:
+    _import_structure["beta_sigmas"] = ["BetaSigmas"]
+
+
 if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
     try:
         if not is_torch_available():
@@ -47,10 +59,17 @@
     except OptionalDependencyNotAvailable:
         from ...utils.dummy_pt_objects import *  # noqa F403
     else:
-        from .beta_sigmas import BetaSigmas
         from .exponential_sigmas import ExponentialSigmas
         from .karras_sigmas import KarrasSigmas
 
+    try:
+        if not (is_torch_available() and is_scipy_available()):
+            raise OptionalDependencyNotAvailable()
+    except OptionalDependencyNotAvailable:
+        from ...utils.dummy_torch_and_scipy_objects import *  # noqa F403
+    else:
+        from .beta_sigmas import BetaSigmas
+
 
 else:
     import sys

src/diffusers/schedulers/sigmas/beta_sigmas.py

@@ -40,6 +40,8 @@ def __init__(
         self.beta = beta
 
     def __call__(self, in_sigmas: torch.Tensor):
+        if not is_scipy_available():
+            raise ImportError("Make sure to install scipy if you want to use beta sigmas.")
         sigma_min = self.sigma_min
         if sigma_min is None:
             sigma_min = in_sigmas[-1].item()

src/diffusers/utils/dummy_torch_and_scipy_objects.py

@@ -15,3 +15,9 @@ def from_config(cls, *args, **kwargs):
     @classmethod
     def from_pretrained(cls, *args, **kwargs):
         requires_backends(cls, ["torch", "scipy"])
+
+class BetaSigmas(metaclass=DummyObject):
+    _backends = ["torch", "scipy"]
+
+    def __init__(self, *args, **kwargs):
+        requires_backends(self, ["torch", "scipy"])