18 lines
585 B
Python
18 lines
585 B
Python
from .ConditioningMathNode import ConditioningMathNode
|
|
from .LatentMathNode import LatentMathNode
|
|
from .ImageMathNode import ImageMathNode
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"mrmth_ConditioningMathNode": ConditioningMathNode,
|
|
"mrmth_LatentMathNode": LatentMathNode,
|
|
"mrmth_ImageMathNode": ImageMathNode
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"mrmth_ConditioningMathNode": "Conditioning math node",
|
|
"mrmth_LatentMathNode": "Latent math node",
|
|
"mrmth_ImageMathNode": "Image math node",
|
|
"Tensor": "Tensor expression",
|
|
"pooled_output": "Pooled output tensor expression"
|
|
}
|