13 lines
321 B
Python
13 lines
321 B
Python
from .nodes.apply_ref_unet_node import ApplyRefUNetNode
|
|
from .nodes.ref_apply_node import RefApplyNode
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"ApplyRefUNetNode": ApplyRefUNetNode,
|
|
"RefApplyNode": RefApplyNode,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"ApplyRefUNetNode": "Apply Ref UNet",
|
|
"RefApplyNode": "Apply Reference",
|
|
}
|