From d6ccd5bc8179ad05804301e587c4760dbf52c139 Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Fri, 5 Apr 2024 07:58:05 +1100 Subject: [PATCH] feat(nodes): disable mosaic fill Needs a bit of tweaking, leaving the code in just disabled/commented it out. --- invokeai/app/api/routers/app_info.py | 2 +- invokeai/app/invocations/infill.py | 10 +++++----- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/invokeai/app/api/routers/app_info.py b/invokeai/app/api/routers/app_info.py index fadc749c3b..21286ac2b0 100644 --- a/invokeai/app/api/routers/app_info.py +++ b/invokeai/app/api/routers/app_info.py @@ -100,7 +100,7 @@ async def get_app_deps() -> AppDependencyVersions: @app_router.get("/config", operation_id="get_config", status_code=200, response_model=AppConfig) async def get_config() -> AppConfig: - infill_methods = ["tile", "lama", "cv2", "color", "mosaic"] + infill_methods = ["tile", "lama", "cv2", "color"] # TODO: add mosaic back if PatchMatch.patchmatch_available(): infill_methods.append("patchmatch") diff --git a/invokeai/app/invocations/infill.py b/invokeai/app/invocations/infill.py index 742ea6a121..418bc62fdc 100644 --- a/invokeai/app/invocations/infill.py +++ b/invokeai/app/invocations/infill.py @@ -22,9 +22,9 @@ logger = InvokeAILogger.get_logger() def get_infill_methods(): - methods = Literal["tile", "color", "lama", "cv2", "mosaic"] + methods = Literal["tile", "color", "lama", "cv2"] # TODO: add mosaic back if PatchMatch.patchmatch_available(): - methods = Literal["patchmatch", "tile", "color", "lama", "cv2", "mosaic"] + methods = Literal["patchmatch", "tile", "color", "lama", "cv2"] # TODO: add mosaic back return methods @@ -145,9 +145,9 @@ class CV2InfillInvocation(InfillImageProcessorInvocation): return cv2_inpaint(image) -@invocation( - "infill_mosaic", title="Mosaic Infill", tags=["image", "inpaint", "outpaint"], category="inpaint", version="1.0.0" -) +# @invocation( +# "infill_mosaic", title="Mosaic Infill", tags=["image", "inpaint", "outpaint"], category="inpaint", version="1.0.0" +# ) class MosaicInfillInvocation(InfillImageProcessorInvocation): """Infills transparent areas of an image with a mosaic pattern drawing colors from the rest of the image"""