init

2024-03-28 18:42:47 -06:00
commit fd802682ab
11 changed files with 407 additions and 0 deletions
@@ -0,0 +1,14 @@
+from .nodes.apply_ref_unet_node import ApplyRefUNetNode
+from .nodes.ref_sampler_node import RefSamplerNode, RefSamplerCustomNode
+
+NODE_CLASS_MAPPINGS = {
+    "ApplyRefUNetNode": ApplyRefUNetNode,
+    "RefSamplerNode": RefSamplerNode,
+    "RefSamplerCustomNode": RefSamplerCustomNode,
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "ApplyRefUNetNode": "Apply Ref UNet",
+    "RefSamplerNode": "Ref Sampler",
+    "RefSamplerCustomNode": "Ref Sampler Custom",
+}