diff --git a/api/onnx_web/image/ade_palette.py b/api/onnx_web/image/ade_palette.py index da781a5f..c4e213a3 100644 --- a/api/onnx_web/image/ade_palette.py +++ b/api/onnx_web/image/ade_palette.py @@ -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 diff --git a/api/onnx_web/image/source_filter.py b/api/onnx_web/image/source_filter.py index 71d8ed07..f0acceb3 100644 --- a/api/onnx_web/image/source_filter.py +++ b/api/onnx_web/image/source_filter.py @@ -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( diff --git a/api/pyproject.toml b/api/pyproject.toml index 53c7d5bf..b863bbfb 100644 --- a/api/pyproject.toml +++ b/api/pyproject.toml @@ -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 diff --git a/api/setup.cfg b/api/setup.cfg index c006aef5..2ab5609b 100644 --- a/api/setup.cfg +++ b/api/setup.cfg @@ -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 \ No newline at end of file + onnx_web/diffusers/pipelines/lpw.py \ No newline at end of file