update name

This commit is contained in:
phamhungd
2026-02-03 14:03:36 +07:00
parent 1f084e9304
commit 0abb5f0698
2 changed files with 19 additions and 19 deletions
+1 -19
View File
@@ -1,24 +1,6 @@
from .node import *
from .install import *
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 = {
'🧩 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']
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
+18
View File
@@ -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',
}