Add LoRA stacks with per-item strengths and named video/image generation presets so setups survive reload without mixing Klein onto MiniMax. Keep LTX Pink Cherry and xAIGen LoRAs on xAIGen, and run Recommend on Beast.

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

server/api/extend.post.ts

@@ -1,6 +1,9 @@
 import { copyFileSync, existsSync, readFileSync } from 'node:fs'
 import { join } from 'node:path'
 import { composeShotPrompt } from '~/utils/globalLocks'
+import { parsePostedLoraStack, listStudioLoras, persistLoraFields } from '~/server/utils/loras'
+import { readLoraStack } from '~/utils/loras'
+import { defaultVideoSteps, isLtxWorkflow, LTX_DISABLED_MESSAGE, ltxWorkflowEnabled, parseVideoWorkflow } from '~/utils/videoModels'
 
 const SAMPLERS = new Set(['res_multistep', 'euler', 'dpmpp_2m'])
 const SCHEDULERS = new Set(['simple', 'ddim_uniform', 'sgm_uniform'])
@@ -39,7 +42,7 @@ function parseScheduler(raw: unknown) {
 }
 
 export default defineEventHandler(async (event) => {
-  const body = await readBody<{ clipId?: string; prompt?: string; duration?: number }>(event).catch(() => ({}))
+  const body = await readBody<{ clipId?: string; prompt?: string; duration?: number; lora?: unknown; loraStack?: unknown }>(event).catch(() => ({}))
   const clipId = String(body?.clipId || '').trim()
   const prompt = String(body?.prompt || '').trim()
   if (!clipId) {
@@ -66,13 +69,21 @@ export default defineEventHandler(async (event) => {
     shotIndex: (source.chainIndex || 0) + 1,
     familyRefs: source.permanenceRefs
   })
+  const workflow = parseVideoWorkflow(source.workflow)
+  if (isLtxWorkflow(workflow) && !ltxWorkflowEnabled()) {
+    throw createError({ statusCode: 400, statusMessage: LTX_DISABLED_MESSAGE })
+  }
   const turbo = source.turbo !== false
   const fps = parseFps(source.fps)
   const cfg = parseCfg(source.cfg, turbo)
   const samplerName = parseSampler(source.samplerName)
   const scheduler = parseScheduler(source.scheduler)
-  const steps = Number(source.steps || defaultVideoSteps(turbo, parseVideoWorkflow(source.workflow)))
+  const steps = Number(source.steps || defaultVideoSteps(turbo, workflow))
   const seed = Math.floor(Math.random() * 2_147_483_647)
+  await listStudioLoras()
+  const postedStack = parsePostedLoraStack(body?.loraStack ?? body?.lora, 'video')
+  const loraStack = postedStack.length ? postedStack : parsePostedLoraStack(readLoraStack(source), 'video')
+  const loraFields = persistLoraFields(loraStack)
   const length = frameLength(durationSeconds, fps)
   const width = source.width || 1344
   const height = source.height || 768
@@ -109,9 +120,10 @@ export default defineEventHandler(async (event) => {
     familyId: source.familyId || crypto.randomUUID(),
     parentClipId: source.id,
     chainIndex: (source.chainIndex || 0) + 1,
-    workflow: parseVideoWorkflow(source.workflow),
+    workflow,
     globalLocks: source.globalLocks,
-    permanenceRefs: source.permanenceRefs
+    permanenceRefs: source.permanenceRefs,
+    ...loraFields
   }
 
   emitJob(job, { type: 'status', message: 'Extracting last frame & stitching extension...', progress: 2 })
@@ -130,7 +142,8 @@ export default defineEventHandler(async (event) => {
     cfg,
     fps,
     samplerName,
-    scheduler
+    scheduler,
+    ...loraFields
   }).catch((error) => {
     removeExtendTemp(job.library?.extendTmpDir)
     const message = error instanceof Error ? error.message : String(error)
@@ -169,6 +182,8 @@ async function runExtend(
     fps: number
     samplerName: string
     scheduler: string
+    loraName?: string
+    loraStack?: import('~/utils/loras').LoraStackItem[]
   }
 ) {
   copyFileSync(params.sourcePath, params.part1Path)
@@ -218,7 +233,8 @@ async function runExtend(
     filenamePrefix: comfyFilenamePrefix(),
     sound,
     workflow: job.library?.workflow,
-    duration: job.library?.duration
+    duration: job.library?.duration,
+    loraStack: params.loraStack || job.library?.loraStack || params.loraName || job.library?.loraName
   })
 
   const queued = await queuePrompt(graph, job.clientId)
@@ -254,7 +270,9 @@ async function runExtend(
       workflow: job.library.workflow,
       duration: job.library.duration,
       globalLocks: job.library.globalLocks,
-      permanenceRefs: job.library.permanenceRefs
+      permanenceRefs: job.library.permanenceRefs,
+      loraName: job.library.loraName,
+      loraStack: job.library.loraStack
     })
   }
   emitJob(job, { type: 'status', message: 'Job queued on ComfyUI', progress: 8 })