check_torch_dependencies

Author: hlky

src/diffusers/schedulers/scheduling_heun_discrete.py

@@ -22,7 +22,11 @@
 from ..configuration_utils import ConfigMixin, register_to_config
 from ..utils import BaseOutput, is_scipy_available
 from .scheduling_utils import KarrasDiffusionSchedulers, SchedulerMixin
-from .sigmas import BetaSigmas, ExponentialSigmas, KarrasSigmas
+from .sigmas import ExponentialSigmas, KarrasSigmas
+
+
+if is_scipy_available():
+    from .sigmas import BetaSigmas
 
 
 @dataclass