diff --git a/src/more_math/AudioMathNode.py b/src/more_math/AudioMathNode.py index aa6cfbf..5316aeb 100644 --- a/src/more_math/AudioMathNode.py +++ b/src/more_math/AudioMathNode.py @@ -31,6 +31,7 @@ class AudioMathNode(io.ComfyNode): return io.Schema( node_id="mrmth_AudioMathNode", category="More math", + display_name="Audio math", inputs=[ io.Audio.Input(id="a", tooltip="Input audio tensor"), io.Audio.Input(id="b", optional=True, tooltip="Second input audio tensor"), diff --git a/src/more_math/ConditioningMathNode.py b/src/more_math/ConditioningMathNode.py index 00e0344..0e58e92 100644 --- a/src/more_math/ConditioningMathNode.py +++ b/src/more_math/ConditioningMathNode.py @@ -17,9 +17,9 @@ class ConditioningMathNode(io.ComfyNode): w, x, y, z: Floats, bound to variables of the expression. Defaults to 0.0 if not provided. Tensor expression: - String, describing expression to mix tensor part of conditioning. Tensor probably describes the composition of the image as it has the say what is on the image. Valid functions are sin, cos, tan, abs, sqrt, min, max, norm. Valid operators are +, -, *, /, ^, %. Usable constants are e and pi. + String, describing expression to mix tensor part of conditioning. Tensor probably describes the composition of the image as it has the say what is on the image. Pooled output expression: - String, describing expression to mix pooled_output part of conditioning. pooled_output does not have that much say in the image but can change some details of the image or completly change the image in some cases, probably also used for positional and temporal conditioning. Expression uses same syntax as Tensor expression. + String, describing expression to mix pooled_output part of conditioning. pooled_output is Tensor compressed into 1 token. outputs: CONDITIONING: diff --git a/src/more_math/FloatMathNode.py b/src/more_math/FloatMathNode.py index 6f25155..51a515c 100644 --- a/src/more_math/FloatMathNode.py +++ b/src/more_math/FloatMathNode.py @@ -36,6 +36,7 @@ class FloatMathNode(io.ComfyNode): return io.Schema( node_id="mrmth_FloatMathNode", category="More math", + display_name="Float math", inputs=[ io.Float.Input(id="a", force_input=True), io.Float.Input(id="b", default=0.0,optional=True, force_input=True), diff --git a/src/more_math/ImageMathNode.py b/src/more_math/ImageMathNode.py index 3127b7e..b7a50d1 100644 --- a/src/more_math/ImageMathNode.py +++ b/src/more_math/ImageMathNode.py @@ -20,7 +20,7 @@ class ImageMathNode(io.ComfyNode): w, x, y, z: Floats, bound to variables of the expression. Defaults to 0.0 if not provided. Image expression: - String, describing expression to mix images. Valid functions are sin, cos, tan, asin, acos, atan, atan2, sinh, cosh, tanh, asinh, acosh, atanh, abs, sqrt, ln, log, exp, pow, min, max, norm, floor, ceil, round, gamma. Valid operators are +, -, *, /, %, ^,!˛&,|. Usable constants are e and pi. + String, describing expression to mix images. outputs: LATENT: @@ -36,6 +36,7 @@ class ImageMathNode(io.ComfyNode): return io.Schema( node_id="mrmth_ImageMathNode", category="More math", + display_name="Image math", inputs=[ io.Image.Input(id="a"), io.Image.Input(id="b", optional=True), diff --git a/src/more_math/LatentMathNode.py b/src/more_math/LatentMathNode.py index aca1feb..cc2779c 100644 --- a/src/more_math/LatentMathNode.py +++ b/src/more_math/LatentMathNode.py @@ -35,6 +35,7 @@ class LatentMathNode(io.ComfyNode): """ return io.Schema( node_id="mrmth_LatentMathNode", + display_name="Latent math", category="More math", inputs=[ io.Latent.Input(id="a"), diff --git a/src/more_math/nodes.py b/src/more_math/nodes.py index 0e173dd..134a8a3 100644 --- a/src/more_math/nodes.py +++ b/src/more_math/nodes.py @@ -19,6 +19,7 @@ class IntToFloatNode(io.ComfyNode): """ return io.Schema( node_id="mrmth_IntToFloat", + display_name="Int -> Float", category="More math", inputs=[ io.Int.Input(id="value", default=0), @@ -42,6 +43,7 @@ class FloatToIntNode(io.ComfyNode): return io.Schema( node_id="mrmth_FloatToInt", category="More math", + display_name="Float -> Int", inputs=[ io.Float.Input(id="value", default=0.0), ], @@ -54,22 +56,6 @@ class FloatToIntNode(io.ComfyNode): return (int(value),) -NODE_id_MAPPINGS = { - "mrmth_ConditioningMathNode": "Conditioning math", - "mrmth_LatentMathNode": "Latent math", - "mrmth_ImageMathNode": "Image math", - "mrmth_FloatMathNode": "Float math", - "mrmth_NoiseMathNode": "Noise math", - "mrmth_AudioMathNode": "Audio math", - "mrmth_IntToFloat": "Int → Float", - "mrmth_FloatToInt": "Float → Int", - "Tensor": "Tensor expression", - "Latent": "Latent expression", - "Image": "Image expression", - "FloatFunc": "Float expression", - "pooled_output": "Pooled output tensor expression" -} - class MoreMathExtension(ComfyExtension): def __init__(self): pass