33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from .brushnet_nodes import BrushNetLoader, BrushNet, BlendInpaint, PowerPaintCLIPLoader, PowerPaint, CutForInpaint
|
|
from .raunet_nodes import RAUNet
|
|
|
|
"""
|
|
@author: nullquant
|
|
@title: BrushNet
|
|
@nickname: BrushName nodes
|
|
@description: These are custom nodes for ComfyUI native implementation of BrushNet, PowerPaint and RAUNet models
|
|
"""
|
|
|
|
# A dictionary that contains all nodes you want to export with their names
|
|
# NOTE: names should be globally unique
|
|
NODE_CLASS_MAPPINGS = {
|
|
"BrushNetLoader": BrushNetLoader,
|
|
"BrushNet": BrushNet,
|
|
"BlendInpaint": BlendInpaint,
|
|
"PowerPaintCLIPLoader": PowerPaintCLIPLoader,
|
|
"PowerPaint": PowerPaint,
|
|
"CutForInpaint": CutForInpaint,
|
|
"RAUNet": RAUNet,
|
|
}
|
|
|
|
# A dictionary that contains the friendly/humanly readable titles for the nodes
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"BrushNetLoader": "BrushNet Loader",
|
|
"BrushNet": "BrushNet",
|
|
"BlendInpaint": "Blend Inpaint",
|
|
"PowerPaintCLIPLoader": "PowerPaint CLIP Loader",
|
|
"PowerPaint": "PowerPaint",
|
|
"CutForInpaint": "Cut For Inpaint",
|
|
"RAUNet": "RAUNet",
|
|
}
|