51 lines
2.1 KiB
Python
51 lines
2.1 KiB
Python
from .node.color_generator import NODE_CLASS_MAPPINGS as COLOR_NODES
|
|
from .node.dimension_calculator import NODE_CLASS_MAPPINGS as DIMENSION_NODES
|
|
from .node.image_to_video import NODE_CLASS_MAPPINGS as VIDEO_NODES
|
|
from .node.advanced_fusion import NODE_CLASS_MAPPINGS as FUSION_NODES
|
|
from .node.simple_latent import NODE_CLASS_MAPPINGS as LATENT_NODES
|
|
from .node.model_optimizer import NODE_CLASS_MAPPINGS as OPTIMIZER_NODES
|
|
from .node.sampling_optimizer import NODE_CLASS_MAPPINGS as SAMPLING_NODES
|
|
from .node.image_resize import NODE_CLASS_MAPPINGS as RESIZE_NODES
|
|
from .node.mask_resize import NODE_CLASS_MAPPINGS as MASK_RESIZE_NODES
|
|
from .node.image_size_limiter import NODE_CLASS_MAPPINGS as SIZE_LIMITER_NODES
|
|
from .node.model_switcher import NODE_CLASS_MAPPINGS as MODEL_SWITCHER_NODES
|
|
from .node.condition_switcher import NODE_CLASS_MAPPINGS as CONDITION_SWITCHER_NODES
|
|
from .node.latent_switcher import NODE_CLASS_MAPPINGS as LATENT_SWITCHER_NODES
|
|
|
|
# 节点类映射
|
|
NODE_CLASS_MAPPINGS = {
|
|
**COLOR_NODES,
|
|
**DIMENSION_NODES,
|
|
**VIDEO_NODES,
|
|
**FUSION_NODES,
|
|
**LATENT_NODES,
|
|
**OPTIMIZER_NODES,
|
|
**SAMPLING_NODES,
|
|
**RESIZE_NODES,
|
|
**MASK_RESIZE_NODES,
|
|
**SIZE_LIMITER_NODES,
|
|
**MODEL_SWITCHER_NODES,
|
|
**CONDITION_SWITCHER_NODES,
|
|
**LATENT_SWITCHER_NODES,
|
|
}
|
|
|
|
# 节点显示名称映射（使用英文作为默认，中文通过locales提供）
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"DD-ColorBackgroundGenerator": "DD Color Background Generator",
|
|
"DD-DimensionCalculator": "DD Dimension Calculator",
|
|
"DD-ImageToVideo": "DD Image To Video",
|
|
"DD-AdvancedFusion": "DD Advanced Fusion",
|
|
"DD-SimpleLatent": "DD Simple Latent",
|
|
"DD-ModelOptimizer": "DD Model Optimizer",
|
|
"DD-SamplingOptimizer": "DD Sampling Optimizer",
|
|
"DD-ImageUniformSize": "DD Image Uniform Size",
|
|
"DD-MaskUniformSize": "DD Mask Uniform Size",
|
|
"DD-ImageSizeLimiter": "DD Image Size Limiter",
|
|
"DD-ModelSwitcher": "DD Model Switcher",
|
|
"DD-ConditionSwitcher": "DD Condition Switcher",
|
|
"DD-LatentSwitcher": "DD Latent Switcher",
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|
|
|
|
WEB_DIRECTORY = "extensions" |