From a5bc21cf50162d0f6a827b9585350ccdfc29406c Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Fri, 11 Apr 2025 15:03:20 +1000 Subject: [PATCH] feat(nodes): extract LaMa model url to constant --- invokeai/app/invocations/infill.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/invokeai/app/invocations/infill.py b/invokeai/app/invocations/infill.py index 3314d72620..dd7b2c87b0 100644 --- a/invokeai/app/invocations/infill.py +++ b/invokeai/app/invocations/infill.py @@ -127,13 +127,16 @@ class InfillPatchMatchInvocation(InfillImageProcessorInvocation): return infilled +LAMA_MODEL_URL = "https://github.com/Sanster/models/releases/download/add_big_lama/big-lama.pt" + + @invocation("infill_lama", title="LaMa Infill", tags=["image", "inpaint"], category="inpaint", version="1.2.2") class LaMaInfillInvocation(InfillImageProcessorInvocation): """Infills transparent areas of an image using the LaMa model""" def infill(self, image: Image.Image): with self._context.models.load_remote_model( - source="https://github.com/Sanster/models/releases/download/add_big_lama/big-lama.pt", + source=LAMA_MODEL_URL, loader=LaMA.load_jit_model, ) as model: lama = LaMA(model)