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