From f65472f426cf4329cd5fdb7be8ea26507f2d24c8 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Tue, 24 Mar 2026 01:44:31 +0000
Subject: [PATCH 1/2] Initial plan


From 934a2cf3b94e324e70637360d29bd3423b4e4a5d Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Tue, 24 Mar 2026 01:49:18 +0000
Subject: [PATCH 2/2] Use chained Collect nodes for FLUX.2 reference images to
 ensure deterministic ordering

Co-authored-by: lstein <111189+lstein@users.noreply.github.com>
Agent-Logs-Url: https://github.com/lstein/InvokeAI/sessions/16c1c72d-c231-4c95-b286-918b26ccafdf
---
 .../util/graph/generation/buildFLUXGraph.ts    | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/invokeai/frontend/web/src/features/nodes/util/graph/generation/buildFLUXGraph.ts b/invokeai/frontend/web/src/features/nodes/util/graph/generation/buildFLUXGraph.ts
index db7cba5961a..ba27e5dbf6e 100644
--- a/invokeai/frontend/web/src/features/nodes/util/graph/generation/buildFLUXGraph.ts
+++ b/invokeai/frontend/web/src/features/nodes/util/graph/generation/buildFLUXGraph.ts
@@ -271,10 +271,7 @@ export const buildFLUXGraph = async (arg: GraphBuilderArg): Promise<GraphBuilder
       .filter((entity) => getGlobalReferenceImageWarnings(entity, model).length === 0);
 
     if (validFlux2RefImageConfigs.length > 0) {
-      const flux2KontextCollect = g.addNode({
-        type: 'collect',
-        id: getPrefixedId('flux2_kontext_collect'),
-      });
+      let prevCollect: Invocation<'collect'> | null = null;
       for (const { config } of validFlux2RefImageConfigs) {
         // FLUX.2 uses the same flux_kontext node - it just packages the image
         const kontextConditioning = g.addNode({
@@ -282,9 +279,18 @@ export const buildFLUXGraph = async (arg: GraphBuilderArg): Promise<GraphBuilder
           id: getPrefixedId('flux_kontext'),
           image: zImageField.parse(config.image?.crop?.image ?? config.image?.original.image),
         });
-        g.addEdge(kontextConditioning, 'kontext_cond', flux2KontextCollect, 'item');
+        const collectNode = g.addNode({
+          type: 'collect',
+          id: getPrefixedId('flux2_kontext_collect'),
+        });
+        g.addEdge(kontextConditioning, 'kontext_cond', collectNode, 'item');
+        if (prevCollect !== null) {
+          g.addEdge(prevCollect, 'collection', collectNode, 'collection');
+        }
+        prevCollect = collectNode;
       }
-      g.addEdge(flux2KontextCollect, 'collection', flux2Denoise, 'kontext_conditioning');
+      assert(prevCollect !== null);
+      g.addEdge(prevCollect, 'collection', flux2Denoise, 'kontext_conditioning');
 
       g.upsertMetadata({ ref_images: validFlux2RefImageConfigs }, 'merge');
     }