23 lines
973 B
Python
23 lines
973 B
Python
from .nodes import (EraserNode, GenFillNode, ShotByTextNode, ShotByImageNode, TailoredGenNode,
|
|
TailoredModelInfoNode, Text2ImageBaseNode)
|
|
# Map the node class to a name used internally by ComfyUI
|
|
NODE_CLASS_MAPPINGS = {
|
|
"BriaEraser": EraserNode, # Return the class, not an instance
|
|
"BriaGenFill": GenFillNode,
|
|
"ShotByTextNode": ShotByTextNode,
|
|
"ShotByImageNode": ShotByImageNode,
|
|
"BriaTailoredGen": TailoredGenNode,
|
|
"TailoredModelInfoNode": TailoredModelInfoNode,
|
|
"Text2ImageBaseNode": Text2ImageBaseNode,
|
|
}
|
|
# Map the node display name to the one shown in the ComfyUI node interface
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"BriaEraser": "Bria Eraser",
|
|
"BriaGenFill": "Bria GenFill",
|
|
"ShotByTextNode": "Bria Shot By Text",
|
|
"ShotByImageNode": "Bria Shot By Image",
|
|
"BriaTailoredGen": "Bria Tailored Gen",
|
|
"TailoredModelInfoNode": "Bria Tailored Model Info",
|
|
"Text2ImageBaseNode": "Bria Text2Image Base",
|
|
}
|