From a395bbf3b4bd7ae6b7c97ce01512bdfa5c10f2ba Mon Sep 17 00:00:00 2001 From: mcDandy Date: Sun, 25 Jan 2026 22:21:15 +0100 Subject: [PATCH] Update ImageMathNode.py --- more_math/ImageMathNode.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/more_math/ImageMathNode.py b/more_math/ImageMathNode.py index 0d1dcef..14bb196 100644 --- a/more_math/ImageMathNode.py +++ b/more_math/ImageMathNode.py @@ -129,17 +129,17 @@ class ImageMathNode(io.ComfyNode): "channel_count": ae.shape[3], } | generate_dim_variables(ae) - norm_v = normalize_to_common_shape(ae, v, mode=length_mismatch)[1] # Add all dynamic inputs for k, v in V.items(): if v is not None: # Normalize all images in V to match ae.shape # Note: normalize_to_common_shape args are *tensors. # We normalize individual V item against 'ae' (the reference shape) + norm_v = normalize_to_common_shape(ae, v, mode=length_mismatch)[1] variables[k] = norm_v - for k, v in F.items(): - variables[k] = v if v is not None else 0.0 + for k, val in F.items(): + variables[k] = val if val is not None else 0.0 tree = parse_expr(Expression); visitor = UnifiedMathVisitor(variables, ae.shape)