diff --git a/api/onnx_web/diffusers/version_safe_diffusers.py b/api/onnx_web/diffusers/version_safe_diffusers.py index 3d51ea4b..8c4da406 100644 --- a/api/onnx_web/diffusers/version_safe_diffusers.py +++ b/api/onnx_web/diffusers/version_safe_diffusers.py @@ -14,7 +14,7 @@ except ImportError: try: from diffusers import DPMSolverSDEScheduler -except: +except ImportError: from ..diffusers.stub_scheduler import StubScheduler as DPMSolverSDEScheduler try: