apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply lint

This commit is contained in:
Sean Sube 2023-04-15 12:13:25 -05:00
parent ac9b3d3b08
commit 43d545e524
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
4 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/onnx_web/image/ade_palette.py
View File

@ -1,6 +1,7 @@
# same palette found in https://github.com/huggingface/diffusers/blob/53377ef83c6446033f3ee506e3ef718db817b293/examples/community/stable_diffusion_controlnet_inpaint_img2img.py#L38
# and https://github.com/ForserX/StableDiffusionUI/blob/main/data/repo/diffusion_scripts/modules/controlnet/palette.py
# and https://github.com/azuritecoin/OnnxDiffusersUI/blob/main/modules/controlnet_utils.py
# same palette found in:
# - https://github.com/huggingface/diffusers/blob/main/examples/community/stable_diffusion_controlnet_inpaint_img2img.py#L38
# - https://github.com/ForserX/StableDiffusionUI/blob/main/data/repo/diffusion_scripts/modules/controlnet/palette.py
# - https://github.com/azuritecoin/OnnxDiffusersUI/blob/main/modules/controlnet_utils.py
# and others

4
api/onnx_web/image/source_filter.py
View File

@ -14,6 +14,7 @@ from PIL import Image, ImageChops, ImageFilter
from ..server.context import ServerContext
from .ade_palette import ade_palette
from .laion_face import generate_annotation
from .noise_source import noise_source_histogram
logger = getLogger(__name__)
@ -29,8 +30,9 @@ def filter_model_path(server: ServerContext, filter_name: str) -> str:
def source_filter_gaussian(
server: ServerContext,
source: Image.Image,
kernel: int = 3,
):
return source.filter(ImageFilter.GaussianBlur(3))
return source.filter(ImageFilter.GaussianBlur(kernel))
def source_filter_noise(

4
api/pyproject.toml
View File

@ -1,10 +1,10 @@
[tool.black]
force-exclude = '''/(diffusion_original|lpw_stable_diffusion_onnx|pipeline_onnx_stable_diffusion_upscale).py'''
force-exclude = '''/(diffusion_original|lpw).py'''
[tool.isort]
profile = "black"
force_to_top = ".logging"
skip_glob = ["*/lpw_stable_diffusion_onnx.py", "*/pipeline_onnx_stable_diffusion_upscale.py"]
skip_glob = ["*/lpw.py"]
[tool.mypy]
# ignore_missing_imports = true

3
api/setup.cfg
View File

@ -6,5 +6,4 @@ ignore = E203, W503
max-line-length = 160
per-file-ignores = __init__.py:F401
exclude =
onnx_web/diffusers/lpw_stable_diffusion_onnx.py
onnx_web/diffusers/pipeline_onnx_stable_diffusion_upscale.py
onnx_web/diffusers/pipelines/lpw.py