apply lint
This commit is contained in:
parent
95841ffe2b
commit
317029356e
|
@ -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__)
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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
|
Loading…
Reference in New Issue