19 lines
485 B
Python
Executable File
19 lines
485 B
Python
Executable File
from .node import *
|
|
import os.path
|
|
|
|
custom_nodes_path = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
'SemanticSAMLoader': SemanticSAMLoader,
|
|
'SemanticSAMSegment': SemanticSAMSegment,
|
|
'PointPrompt': PointPrompt
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"SemanticSAMLoader": "Semantic SAM Model Loader",
|
|
"SemanticSAMSegment": "Semantic SAM Segment",
|
|
"PointPrompt" : "Point Prompt"
|
|
}
|
|
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] |