16 lines
505 B
Python
16 lines
505 B
Python
from .nodes.apply_ref_unet_node import ApplyRefUNetNode
|
|
from .nodes.apply_ref_content_node import ApplyRefContentNode
|
|
from .nodes.apply_ref_style_node import ApplyRefStyleNode
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"ApplyRefUNetNode": ApplyRefUNetNode,
|
|
"ApplyRefContentNode": ApplyRefContentNode,
|
|
"ApplyRefStyleNode": ApplyRefStyleNode,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"ApplyRefUNetNode": "Apply Ref UNet",
|
|
"ApplyRefContentNode": "Apply Ref Content",
|
|
"ApplyRefStyleNode": "Apply Ref Style",
|
|
}
|