From 1f084e93042c99a59a192cc24ec6770ce2c1fdaf Mon Sep 17 00:00:00 2001 From: phamhungd Date: Tue, 3 Feb 2026 13:55:25 +0700 Subject: [PATCH] Update __init__.py --- __init__.py | 21 +++++++++++++++------ 1 file changed, 15 insertions(+), 6 deletions(-) diff --git a/__init__.py b/__init__.py index f0b7a7f..6e0a34d 100644 --- a/__init__.py +++ b/__init__.py @@ -2,12 +2,21 @@ from .node import * from .install import * NODE_CLASS_MAPPINGS = { - '🧩 SAMModel Loader': SAMModelLoader, - '🧩 DinoModel Loader': GroundingDinoModelLoader, - '🧩 Dino SAMSegment': GroundingDinoSAMSegment, - '🧩 Invert Mask': InvertMask, - "🧩 Is Mask Empty": IsMaskEmptyNode, - "🧩 Segment Anything": SegmentAnything, + 'SDVN SAMModelLoader': SAMModelLoader, + 'SDVN DinoModelLoader': GroundingDinoModelLoader, + 'SDVN DinoSAMSegment': GroundingDinoSAMSegment, + 'SDVN InvertMask': InvertMask, + "SDVN IsMaskEmpty": IsMaskEmptyNode, + "SDVN SegmentAnything": SegmentAnything, +} + +NODE_DISPLAY_NAME_MAPPINGS = { + '🧩 SAMModel Loader': 'SDVN SAMModelLoader', + '🧩 DinoModel Loader': 'SDVN DinoModelLoader', + '🧩 Dino SAMSegment': 'SDVN DinoSAMSegment', + '🧩 Invert Mask': 'SDVN InvertMask', + "🧩 Is Mask Empty": 'SDVN IsMaskEmpty', + "🧩 Segment Anything": 'SDVN SegmentAnything', } __all__ = ['NODE_CLASS_MAPPINGS']