24 lines
842 B
Python
24 lines
842 B
Python
from .brushnet_nodes import BrushNetLoader, BrushNetPipeline, BrushNetInpaint, BlendInpaint
|
|
"""
|
|
@author: nullquant
|
|
@title: BrushNet
|
|
@nickname: BrushName nodes
|
|
@description: This repository contains an custom nodes for inpaint using BrushNet and PowerPaint 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,
|
|
"BrushNetPipeline": BrushNetPipeline,
|
|
"BrushNetInpaint": BrushNetInpaint,
|
|
"BlendInpaint": BlendInpaint,
|
|
}
|
|
|
|
# A dictionary that contains the friendly/humanly readable titles for the nodes
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"BrushNetLoader": "BrushNet Loader",
|
|
"BrushNetPipeline": "BrushNet Pipeline",
|
|
"BrushNetInpaint": "BrushNet Inpaint",
|
|
"BlendInpaint": "Blend Inpaint",
|
|
} |