update name

2026-02-03 14:03:36 +07:00
parent 1f084e9304
commit 0abb5f0698
2 changed files with 19 additions and 19 deletions
@@ -427,3 +427,21 @@ class IsMaskEmptyNode:

    def main(self, mask):
        return (torch.all(mask == 0).int().item(), )
+
+NODE_CLASS_MAPPINGS = {
+    'SDVN SAMModelLoader': SAMModelLoader,
+    'SDVN DinoModelLoader': GroundingDinoModelLoader,
+    'SDVN DinoSAMSegment': GroundingDinoSAMSegment,
+    'SDVN InvertMask': InvertMask,
+    "SDVN IsMaskEmpty": IsMaskEmptyNode,
+    "SDVN SegmentAnything": SegmentAnything,
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    'SDVN SAMModelLoader': '🧩 SAMModel Loader',
+    'SDVN DinoModelLoader': '🧩 DinoModel Loader',
+    'SDVN DinoSAMSegment': '🧩 Dino SAMSegment',
+    'SDVN InvertMask': '🧩 Invert Mask',
+    "SDVN IsMaskEmpty": '🧩 Is Mask Empty',
+    "SDVN SegmentAnything": '🧩 Segment Anything',
+}