add device info to visitor and fixed forgotten '

2026-01-30 14:22:13 +01:00
parent a84c4f3048
commit db8000d51e
17 changed files with 1379 additions and 1208 deletions
@@ -137,7 +137,7 @@ class VideoMathNode(io.ComfyNode):
            variables[k] = val if val is not None else 0.0

        tree = parse_expr(Expression);
-        visitor = UnifiedMathVisitor(variables, ae.shape,state_storage=ss)
+        visitor = UnifiedMathVisitor(variables, ae.shape,ae.device,state_storage=ss)
        result = visitor.visit(tree)
        result = as_tensor(result, ae.shape)