Update __init__.py

This commit is contained in:
phamhungd
2026-02-03 13:55:25 +07:00
parent 6c7669041e
commit 1f084e9304
+15 -6
View File
@@ -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']