diff --git a/api/onnx_web/convert/diffusion/control.py b/api/onnx_web/convert/diffusion/control.py
index f2bbbe2b..cc2e4bb5 100644
--- a/api/onnx_web/convert/diffusion/control.py
+++ b/api/onnx_web/convert/diffusion/control.py
@@ -4,9 +4,9 @@ from pathlib import Path
 from typing import Dict
 
 import torch
-from ...diffusers.version_safe_diffusers import AttnProcessor, ControlNetModel
 
 from ...constants import ONNX_MODEL
+from ...diffusers.version_safe_diffusers import AttnProcessor, ControlNetModel
 from ..utils import ConversionContext, is_torch_2_0, onnx_export
 
 logger = getLogger(__name__)
diff --git a/api/onnx_web/diffusers/version_safe_diffusers.py b/api/onnx_web/diffusers/version_safe_diffusers.py
index 0add964b..79198b81 100644
--- a/api/onnx_web/diffusers/version_safe_diffusers.py
+++ b/api/onnx_web/diffusers/version_safe_diffusers.py
@@ -8,23 +8,17 @@ is_diffusers_0_15 = version.parse(
 
 
 try:
-    from diffusers import DEISMultistepScheduler  # NOQA
+    from diffusers import DEISMultistepScheduler
 except ImportError:
-    from ..diffusers.stub_scheduler import (
-        StubScheduler as DEISMultistepScheduler,  # NOQA
-    )
+    from ..diffusers.stub_scheduler import StubScheduler as DEISMultistepScheduler
 
 try:
-    from diffusers import UniPCMultistepScheduler  # NOQA
+    from diffusers import UniPCMultistepScheduler
 except ImportError:
-    from ..diffusers.stub_scheduler import (
-        StubScheduler as UniPCMultistepScheduler,  # NOQA
-    )
+    from ..diffusers.stub_scheduler import StubScheduler as UniPCMultistepScheduler
 
 
 if is_diffusers_0_15:
-    from diffusers.models.attention_processor import AttnProcessor  # NOQA
+    from diffusers.models.attention_processor import AttnProcessor
 else:
-    from diffusers.models.cross_attention import (
-        CrossAttnProcessor as AttnProcessor,  # NOQA
-    )
+    from diffusers.models.cross_attention import CrossAttnProcessor as AttnProcessor
diff --git a/api/setup.cfg b/api/setup.cfg
index 2ab5609b..9cda2544 100644
--- a/api/setup.cfg
+++ b/api/setup.cfg
@@ -4,6 +4,8 @@ description-file = README.md
 [flake8]
 ignore = E203, W503
 max-line-length = 160
-per-file-ignores = __init__.py:F401
+per-file-ignores =
+    __init__.py:F401
+    onnx_web/diffusers/version_safe_diffusers.py:F401
 exclude =
     onnx_web/diffusers/pipelines/lpw.py
\ No newline at end of file