Apply primitive nodes to graph before serializing workflow.

web/scripts/app.js

@@ -1586,6 +1586,16 @@ export class ComfyApp {
 	 * @returns The workflow and node links
 	 */
 	async graphToPrompt() {
+		for (const node of this.graph.computeExecutionOrder(false)) {
+			if (node.isVirtualNode) {
+				// Don't serialize frontend only nodes but let them make changes
+				if (node.applyToGraph) {
+					node.applyToGraph();
+				}
+				continue;
+			}
+		}
+
 		const workflow = this.graph.serialize();
 		const output = {};
 		// Process nodes in order of execution
@@ -1593,10 +1603,6 @@ export class ComfyApp {
 			const n = workflow.nodes.find((n) => n.id === node.id);
 
 			if (node.isVirtualNode) {
-				// Don't serialize frontend only nodes but let them make changes
-				if (node.applyToGraph) {
-					node.applyToGraph(workflow);
-				}
 				continue;
 			}
 

web/scripts/ui.js

@@ -719,7 +719,8 @@ export class ComfyUI {
 							filename += ".json";
 						}
 					}
-					const json = JSON.stringify(app.graph.serialize(), null, 2); // convert the data to a JSON string
+					app.graphToPrompt().then(p=>{
+						const json = JSON.stringify(p.workflow, null, 2); // convert the data to a JSON string
 						const blob = new Blob([json], {type: "application/json"});
 						const url = URL.createObjectURL(blob);
 						const a = $el("a", {
@@ -733,6 +734,7 @@ export class ComfyUI {
 							a.remove();
 							window.URL.revokeObjectURL(url);
 						}, 0);
+					});
 				},
 			}),
 			$el("button", {