Files
aigen/utils/videoModels.ts
T

82 lines
2.8 KiB
TypeScript

export type VideoEngineId = 'minimax' | 'ltx'
export type VideoStartId = 'still' | 'text'
export type MiniMaxGraphId = 'v1' | 'v2'
export type VideoWorkflowId = 'v1' | 'v2' | 't2v' | 'ltx' | 'ltx-t2v'
export function parseVideoWorkflow(raw: unknown): VideoWorkflowId {
const value = String(raw || '').trim()
if (value === 'v2') return 'v2'
if (value === 't2v') return 't2v'
if (value === 'ltx' || value === 'ltx-i2v') return 'ltx'
if (value === 'ltx-t2v') return 'ltx-t2v'
return 'v1'
}
export const LTX_DISABLED_MESSAGE = 'LTX Pink Cherry is not available on this instance'
export function isXaigenStudio(publicConfig?: { studio?: unknown; instanceName?: unknown }) {
let pub = publicConfig
if (!pub) {
try {
pub = useRuntimeConfig().public as { studio?: unknown; instanceName?: unknown }
} catch {
pub = {}
}
}
return /^xaigen$/i.test(String(pub.studio || '').trim())
|| /xaigen/i.test(String(pub.instanceName || '').trim())
}
export function ltxWorkflowEnabled(publicConfig?: { studio?: unknown; instanceName?: unknown }) {
return isXaigenStudio(publicConfig)
}
export function videoEngineOf(id: VideoWorkflowId): VideoEngineId {
return id.startsWith('ltx') ? 'ltx' : 'minimax'
}
export function videoStartOf(id: VideoWorkflowId): VideoStartId {
return id === 't2v' || id === 'ltx-t2v' ? 'text' : 'still'
}
export function isLtxWorkflow(id: VideoWorkflowId) {
return videoEngineOf(id) === 'ltx'
}
export function isTextToVideo(id: VideoWorkflowId) {
return videoStartOf(id) === 'text'
}
export function minimaxGraphOf(id: VideoWorkflowId): MiniMaxGraphId {
return id === 'v2' ? 'v2' : 'v1'
}
export function workflowForExtension(id: VideoWorkflowId): VideoWorkflowId {
if (id === 't2v') return 'v1'
if (id === 'ltx-t2v') return 'ltx'
return id
}
export function composeVideoWorkflow(engine: VideoEngineId, start: VideoStartId, graph: MiniMaxGraphId = 'v1'): VideoWorkflowId {
if (engine === 'ltx') return start === 'text' ? 'ltx-t2v' : 'ltx'
if (start === 'text') return 't2v'
return graph === 'v2' ? 'v2' : 'v1'
}
export function coerceVideoWorkflow(raw: unknown, allowLtx?: boolean): VideoWorkflowId {
const id = parseVideoWorkflow(raw)
if ((allowLtx ?? ltxWorkflowEnabled()) || !isLtxWorkflow(id)) return id
return composeVideoWorkflow('minimax', videoStartOf(id), 'v1')
}
export function defaultVideoSteps(turbo: boolean, workflow: string = 'v1') {
const id = parseVideoWorkflow(workflow)
if (!turbo) return isLtxWorkflow(id) ? 12 : 20
if (isLtxWorkflow(id)) return 8
return id === 'v2' ? 6 : 8
}
export const LTX_UNET = 'PinkCherry_FineTune_int8_v1_8_LTX23.safetensors'
export const LTX_DISTILLED_LORA = 'ltx-2.3-22b-distilled-lora-384-1.1.safetensors'
export const LTX_NEGATIVE = 'blurry, low quality, jitter, deformed, static frame, freeze, watermark, bad anatomy, text, subtitles'