Skip to content

Nodes for better inpainting with ComfyUI: Fooocus inpaint model for SDXL, LaMa, MAT, and various other tools for pre-filling inpaint & outpaint areas.

License

Notifications You must be signed in to change notification settings

henlee1/comfyui-inpaint-nodes

 
 

Repository files navigation

ComfyUI Inpaint Nodes

Nodes for better inpainting with ComfyUI: Fooocus inpaint model for SDXL, LaMa, MAT, and various other tools for pre-filling inpaint & outpaint areas.

Fooocus Inpaint

Adds two nodes which allow using Fooocus inpaint model. It's a small and flexible patch which can be applied to any SDXL checkpoint and will transform it into an inpaint model. This model can then be used like other inpaint models, and provides the same benefits. Read more

Download models from lllyasviel/fooocus_inpaint to ComfyUI/models/inpaint.

Inpaint workflow

Note: Implementation is somewhat hacky as it monkey-patches ComfyUI's ModelPatcher to support the custom Lora format which the model is using.

Inpaint Conditioning

Fooocus inpaint can be used with ComfyUI's VAE Encode (for Inpainting) directly. However this does not allow existing content in the masked area, denoise strength must be 1.0.

InpaintModelConditioning can be used to combine inpaint models with existing content. The resulting latent can however not be used directly to patch the model using Apply Fooocus Inpaint. This repository adds a new node VAE Encode & Inpaint Conditioning which provides two outputs: latent_inpaint (connect this to Apply Fooocus Inpaint) and latent_samples (connect this to KSampler).

It's the same as using both VAE Encode (for Inpainting) and InpaintModelConditioning, but less overhead because it avoids VAE-encoding the image twice. Example workflow

Inpaint Pre-processing

Several nodes are available to fill the masked area prior to inpainting. They avoid seams as long as the input mask is large enough.

Fill Masked

This fills the masked area, with a smooth transition at the border. It has 3 modes:

  • neutral: fills with grey, good for adding entirely new content
  • telea: fills with colors from surrounding border (based on algorithm by Alexandru Telea)
  • navier-stokes: fills with colors from surrounding border (based on fluid dynamics described by Navier-Stokes)
Input Neutral Telea Navier-Stokes
input neutral telea ns

Blur Masked

This blurs the image into the masked area. The blur is less strong at the borders of the mask. Good for keeping the general colors the same.

Input Blur radius 17 Blur radius 65
input blur-17 blur-65

Inpaint Models (LaMA, MAT)

This runs a small, fast inpaint model on the masked area. Models can be loaded with Load Inpaint Model and are applied with the Inpaint (using Model) node. This works well for outpainting or object removal.

The following inpaint models are supported, place them in ComfyUI/models/inpaint:

Input LaMa MAT
input lama mat

Inpaint Post-processing

Denoise to Compositing Mask

Takes a mask, an offset (default 0.1) and a threshold (default 0.2). Maps mask values in the range of [offsetthreshold] to [0 → 1]. Values below offset are clamped to 0, values above threshold to 1.

This is particularly useful in combination with ComfyUI's "Differential Diffusion" node, which allows to use a mask as per-pixel denoise strength. Using the same mask for compositing (alpha blending) defeats the purpose, but no blending at all degrades quality in regions with zero or very low strength. This node creates a mask suitable for blending from the denoise-mask.

Example Workflows

Example workflows can be found in workflows.

  • Simple: basic workflow, ignore previous content, 100% replacement
  • Refine: advanced workflow, refine existing content, 1-100% denoise strength
  • Outpaint: workflow for outpainting with pre-processing
  • Pre-process: complex workflow for experimenting with pre-processors
  • Promptless: same as above but without text prompt, requires IP-Adapter

Installation

Use ComfyUI Manager and search for "ComfyUI Inpaint Nodes".

or download the repository and put the folder into ComfyUI/custom_nodes.

or use GIT:

cd ComfyUI/custom_nodes
git clone https://github.com/Acly/comfyui-inpaint-nodes.git

Restart ComfyUI after installing!


OpenCV is required for telea and navier-stokes fill mode:

pip install opencv-python

Acknowledgements

About

Nodes for better inpainting with ComfyUI: Fooocus inpaint model for SDXL, LaMa, MAT, and various other tools for pre-filling inpaint & outpaint areas.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%