Show Krea in Image v2 sampler progress instead of Flux.2 Klein.

Co-authored-by: Cursor <cursoragent@cursor.com>

server/utils/imageChainV2.ts

@@ -2,7 +2,7 @@ import { createJob, emitJob, type Job } from '~/server/utils/jobs'
 import { ensureComfyReady } from '~/server/utils/comfyLifecycle'
 import { assertImageScaleToTotalPixelsNode, getComfyHost, uploadImage, queuePrompt, purgeComfyArtifacts } from '~/server/utils/comfy'
 import { withImageComfyHost, waitForImageEdit, downloadEditedImage } from '~/server/utils/imageComfy'
-import { buildImageV2Workflow, resolveKreaGenerateAssets } from '~/server/utils/imageWorkflowV2'
+import { buildImageV2Workflow, IMAGE_V2_NODE_LABELS, resolveKreaGenerateAssets } from '~/server/utils/imageWorkflowV2'
 import { ensureComfyLoraNames } from '~/server/utils/loras'
 import { imageDimensions } from '~/server/utils/resolution'
 import { emitChainJob } from '~/server/utils/watch'
@@ -148,6 +148,11 @@ export async function runEditV2(job: Job, params: EditV2RunParams) {
         promptId: queued.prompt_id,
         clientId: job.clientId,
         timeoutMs: 10 * 60 * 1000,
+        engineLabel,
+        nodeLabel: (node) => {
+          if (node === '15' || node === '19') return `Sampling ${engineLabel}`
+          return IMAGE_V2_NODE_LABELS[node]
+        },
         onProgress: (event) => {
           emitChainJob(job, {
             type: 'status',
@@ -209,7 +214,7 @@ export async function runEditV2(job: Job, params: EditV2RunParams) {
       job.status = 'complete'
       emitChainJob(job, {
         type: 'complete',
-        message: 'Klein v2 finished on Beast',
+        message: `${engineLabel} v2 finished on Beast`,
         progress: 100,
         stillId: still?.id,
         filename: output.filename,