apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply lint

This commit is contained in:
Sean Sube 2023-04-15 14:42:14 -05:00
parent 95841ffe2b
commit 317029356e
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
3 changed files with 10 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/convert/diffusion/control.py
View File

@ -4,9 +4,9 @@ from pathlib import Path
from typing import Dict from typing import Dict
import torch import torch
from ...diffusers.version_safe_diffusers import AttnProcessor, ControlNetModel
from ...constants import ONNX_MODEL from ...constants import ONNX_MODEL
from ...diffusers.version_safe_diffusers import AttnProcessor, ControlNetModel
from ..utils import ConversionContext, is_torch_2_0, onnx_export from ..utils import ConversionContext, is_torch_2_0, onnx_export
logger = getLogger(__name__) logger = getLogger(__name__)

18
api/onnx_web/diffusers/version_safe_diffusers.py
View File

@ -8,23 +8,17 @@ is_diffusers_0_15 = version.parse(
try: try:
from diffusers import DEISMultistepScheduler # NOQA from diffusers import DEISMultistepScheduler
except ImportError: except ImportError:
from ..diffusers.stub_scheduler import ( from ..diffusers.stub_scheduler import StubScheduler as DEISMultistepScheduler
StubScheduler as DEISMultistepScheduler, # NOQA
)
try: try:
from diffusers import UniPCMultistepScheduler # NOQA from diffusers import UniPCMultistepScheduler
except ImportError: except ImportError:
from ..diffusers.stub_scheduler import ( from ..diffusers.stub_scheduler import StubScheduler as UniPCMultistepScheduler
StubScheduler as UniPCMultistepScheduler, # NOQA
)
if is_diffusers_0_15: if is_diffusers_0_15:
from diffusers.models.attention_processor import AttnProcessor # NOQA from diffusers.models.attention_processor import AttnProcessor
else: else:
from diffusers.models.cross_attention import ( from diffusers.models.cross_attention import CrossAttnProcessor as AttnProcessor
CrossAttnProcessor as AttnProcessor, # NOQA
)

4
api/setup.cfg
View File

@ -4,6 +4,8 @@ description-file = README.md
[flake8] [flake8]
ignore = E203, W503 ignore = E203, W503
max-line-length = 160 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 = exclude =
onnx_web/diffusers/pipelines/lpw.py onnx_web/diffusers/pipelines/lpw.py