fix(ui): generation_mode metadata not set correctly

Author: psychedelicious

invokeai/frontend/web/src/features/nodes/util/graph/generation/buildCogView4Graph.ts

@@ -94,7 +94,6 @@ export const buildCogView4Graph = async (
   assert(modelConfig.base === 'cogview4');
 
   g.upsertMetadata({
-    generation_mode: 'cogview4_txt2img',
     cfg_scale,
     width: originalSize.width,
     height: originalSize.height,
@@ -111,6 +110,7 @@ export const buildCogView4Graph = async (
 
   if (generationMode === 'txt2img') {
     canvasOutput = addTextToImage({ g, l2i, originalSize, scaledSize });
+    g.upsertMetadata({ generation_mode: 'cogview4_txt2img' });
   } else if (generationMode === 'img2img') {
     canvasOutput = await addImageToImage({
       g,
@@ -125,6 +125,7 @@ export const buildCogView4Graph = async (
       denoising_start,
       fp32: false,
     });
+    g.upsertMetadata({ generation_mode: 'cogview4_img2img' });
   } else if (generationMode === 'inpaint') {
     canvasOutput = await addInpaint({
       state,
@@ -140,6 +141,7 @@ export const buildCogView4Graph = async (
       denoising_start,
       fp32: false,
     });
+    g.upsertMetadata({ generation_mode: 'cogview4_inpaint' });
   } else if (generationMode === 'outpaint') {
     canvasOutput = await addOutpaint({
       state,
@@ -155,6 +157,7 @@ export const buildCogView4Graph = async (
       denoising_start,
       fp32: false,
     });
+    g.upsertMetadata({ generation_mode: 'cogview4_outpaint' });
   } else {
     assert<Equals<typeof generationMode, never>>(false);
   }

invokeai/frontend/web/src/features/nodes/util/graph/generation/buildFLUXGraph.ts

@@ -143,7 +143,6 @@ export const buildFLUXGraph = async (
   addFLUXLoRAs(state, g, denoise, modelLoader, posCond);
 
   g.upsertMetadata({
-    generation_mode: 'flux_txt2img',
     guidance,
     width: originalSize.width,
     height: originalSize.height,
@@ -180,6 +179,7 @@ export const buildFLUXGraph = async (
     });
   } else if (generationMode === 'txt2img') {
     canvasOutput = addTextToImage({ g, l2i, originalSize, scaledSize });
+    g.upsertMetadata({ generation_mode: 'flux_txt2img' });
   } else if (generationMode === 'img2img') {
     canvasOutput = await addImageToImage({
       g,
@@ -194,6 +194,7 @@ export const buildFLUXGraph = async (
       denoising_start,
       fp32: false,
     });
+    g.upsertMetadata({ generation_mode: 'flux_img2img' });
   } else if (generationMode === 'inpaint') {
     canvasOutput = await addInpaint({
       state,
@@ -209,6 +210,7 @@ export const buildFLUXGraph = async (
       denoising_start,
       fp32: false,
     });
+    g.upsertMetadata({ generation_mode: 'flux_inpaint' });
   } else if (generationMode === 'outpaint') {
     canvasOutput = await addOutpaint({
       state,
@@ -224,6 +226,7 @@ export const buildFLUXGraph = async (
       denoising_start,
       fp32: false,
     });
+    g.upsertMetadata({ generation_mode: 'flux_outpaint' });
   } else {
     assert<Equals<typeof generationMode, never>>(false);
   }

invokeai/frontend/web/src/features/nodes/util/graph/generation/buildSD1Graph.ts

@@ -140,7 +140,6 @@ export const buildSD1Graph = async (
   assert(model.base === 'sd-1' || model.base === 'sd-2');
 
   g.upsertMetadata({
-    generation_mode: 'txt2img',
     cfg_scale,
     cfg_rescale_multiplier,
     width: originalSize.width,
@@ -172,6 +171,7 @@ export const buildSD1Graph = async (
 
   if (generationMode === 'txt2img') {
     canvasOutput = addTextToImage({ g, l2i, originalSize, scaledSize });
+    g.upsertMetadata({ generation_mode: 'txt2img' });
   } else if (generationMode === 'img2img') {
     canvasOutput = await addImageToImage({
       g,
@@ -186,6 +186,7 @@ export const buildSD1Graph = async (
       denoising_start,
       fp32: vaePrecision === 'fp32',
     });
+    g.upsertMetadata({ generation_mode: 'img2img' });
   } else if (generationMode === 'inpaint') {
     canvasOutput = await addInpaint({
       state,
@@ -201,6 +202,7 @@ export const buildSD1Graph = async (
       denoising_start,
       fp32: vaePrecision === 'fp32',
     });
+    g.upsertMetadata({ generation_mode: 'inpaint' });
   } else if (generationMode === 'outpaint') {
     canvasOutput = await addOutpaint({
       state,
@@ -216,6 +218,7 @@ export const buildSD1Graph = async (
       denoising_start,
       fp32,
     });
+    g.upsertMetadata({ generation_mode: 'outpaint' });
   } else {
     assert<Equals<typeof generationMode, never>>(false);
   }

invokeai/frontend/web/src/features/nodes/util/graph/generation/buildSD3Graph.ts

@@ -109,7 +109,6 @@ export const buildSD3Graph = async (
   g.addEdge(denoise, 'latents', l2i, 'latents');
 
   g.upsertMetadata({
-    generation_mode: 'sd3_txt2img',
     cfg_scale,
     width: originalSize.width,
     height: originalSize.height,
@@ -136,6 +135,7 @@ export const buildSD3Graph = async (
 
   if (generationMode === 'txt2img') {
     canvasOutput = addTextToImage({ g, l2i, originalSize, scaledSize });
+    g.upsertMetadata({ generation_mode: 'sd3_txt2img' });
   } else if (generationMode === 'img2img') {
     canvasOutput = await addImageToImage({
       g,
@@ -150,6 +150,7 @@ export const buildSD3Graph = async (
       denoising_start,
       fp32: false,
     });
+    g.upsertMetadata({ generation_mode: 'sd3_img2img' });
   } else if (generationMode === 'inpaint') {
     canvasOutput = await addInpaint({
       state,
@@ -165,6 +166,7 @@ export const buildSD3Graph = async (
       denoising_start,
       fp32: false,
     });
+    g.upsertMetadata({ generation_mode: 'sd3_inpaint' });
   } else if (generationMode === 'outpaint') {
     canvasOutput = await addOutpaint({
       state,
@@ -180,6 +182,7 @@ export const buildSD3Graph = async (
       denoising_start,
       fp32: false,
     });
+    g.upsertMetadata({ generation_mode: 'sd3_outpaint' });
   } else {
     assert<Equals<typeof generationMode, never>>(false);
   }

invokeai/frontend/web/src/features/nodes/util/graph/generation/buildSDXLGraph.ts

@@ -140,7 +140,6 @@ export const buildSDXLGraph = async (
   g.addEdge(denoise, 'latents', l2i, 'latents');
 
   g.upsertMetadata({
-    generation_mode: 'sdxl_txt2img',
     cfg_scale,
     cfg_rescale_multiplier,
     width: originalSize.width,
@@ -178,6 +177,7 @@ export const buildSDXLGraph = async (
 
   if (generationMode === 'txt2img') {
     canvasOutput = addTextToImage({ g, l2i, originalSize, scaledSize });
+    g.upsertMetadata({ generation_mode: 'sdxl_txt2img' });
   } else if (generationMode === 'img2img') {
     canvasOutput = await addImageToImage({
       g,
@@ -192,6 +192,7 @@ export const buildSDXLGraph = async (
       denoising_start,
       fp32,
     });
+    g.upsertMetadata({ generation_mode: 'sdxl_img2img' });
   } else if (generationMode === 'inpaint') {
     canvasOutput = await addInpaint({
       state,
@@ -207,6 +208,7 @@ export const buildSDXLGraph = async (
       denoising_start,
       fp32,
     });
+    g.upsertMetadata({ generation_mode: 'sdxl_inpaint' });
   } else if (generationMode === 'outpaint') {
     canvasOutput = await addOutpaint({
       state,
@@ -222,6 +224,7 @@ export const buildSDXLGraph = async (
       denoising_start,
       fp32,
     });
+    g.upsertMetadata({ generation_mode: 'sdxl_outpaint' });
   } else {
     assert<Equals<typeof generationMode, never>>(false);
   }