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 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