towsty/aigen
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Replace Comfy YuE with standalone YuEGP profile 1

Browse Source
This commit is contained in:
Towsty
2026-09-07 20:24:47 -05:00
parent eeff12cdab
commit c2b6032e49
33 changed files with 976 additions and 294 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/utils/musicChain.ts
+8 -13
View File
@@ -1,5 +1,6 @@
import { startYueGpJob } from './yueGp'
import { createJob, emitJob, type Job } from '~/server/utils/jobs'
import { extractAudio, fetchHistory, fetchHistoryAll, findHistoryAudio, freeComfyVram, purgeComfyArtifacts, queuePrompt } from '~/server/utils/comfy'
import { extractAudio, fetchHistory, fetchHistoryAll, findHistoryAudio, purgeComfyArtifacts, queuePrompt } from '~/server/utils/comfy'
import { comfyWsUrl, comfyFetch } from '~/server/utils/comfy'
import { ensureComfyReady } from '~/server/utils/comfyLifecycle'
import { downloadComfyAudio, saveTrack } from '~/server/utils/library'
@@ -21,6 +22,7 @@ export type MusicJobParams = {
lyricsStrength: number
instrumental: boolean
folderLocked?: boolean
yueProfile?: 1 | 3
engine?: MusicEngine
samplerName?: string
scheduler?: string
@@ -33,7 +35,7 @@ function sleep(ms: number) {
function classifyError(message: string) {
const lower = message.toLowerCase()
if (lower.includes('out of memory') || lower.includes('exceed allowed memory') || (lower.includes('cuda') && lower.includes('alloc')) || lower.includes('vram')) {
return 'ComfyUI ran out of VRAM on YuE. Sleep→Poke Comfy, don’t run video/image first, use one short [Chorus] only — or use ACE-Step.'
return 'ComfyUI ran out of VRAM.'
}
if (lower.includes('econnrefused') || lower.includes('unreachable') || lower.includes('fetch failed')) {
return 'ComfyUI host connection dropped. Confirm the desktop instance is running.'
@@ -338,6 +340,7 @@ function watchMusicJob(job: Job): Promise<void> {
}
export async function startMusicJob(params: MusicJobParams) {
if (params.engine === 'yue') return startYueGpJob(params)
const job = createJob('music')
job.library = {
ownerKey: params.ownerKey,
@@ -381,14 +384,10 @@ export async function startMusicJob(params: MusicJobParams) {
})
})
job.status = 'running'
const engineLabel = params.engine === 'yue' ? 'YuE' : params.engine === 'ace-step-1.5' ? 'ACE-Step 1.5' : 'ACE-Step'
const engineLabel = params.engine === 'ace-step-1.5' ? 'ACE-Step 1.5' : 'ACE-Step'
emitJob(job, { type: 'status', message: `Queueing ${engineLabel}...`, progress: 6 })
await waitForSocket(job, 4000)
await assertMusicEngineNodes(params.engine)
if (params.engine === 'yue') {
emitJob(job, { type: 'status', message: 'Freeing Comfy VRAM for YuE…', progress: 5 })
await freeComfyVram()
}
const graph = buildMusicWorkflow({
tags: params.tags,
lyrics: params.instrumental ? '' : params.lyrics,
@@ -406,9 +405,7 @@ export async function startMusicJob(params: MusicJobParams) {
try {
const queued = await queuePrompt(graph, job.clientId, { timeoutMs: 25_000 })
if (queued?.prompt_id) job.promptId = queued.prompt_id
const runningMessage = params.engine === 'yue'
? 'YuE Stage A on Comfy — 16GB job, often 10–20+ minutes. Leave it alone.'
: 'Job queued on ComfyUI'
const runningMessage = 'Job queued on ComfyUI'
emitJob(job, { type: 'status', message: runningMessage, progress: 8 })
} catch (queueError: unknown) {
const statusCode = Number((queueError as { statusCode?: number; status?: number })?.statusCode
@@ -416,9 +413,7 @@ export async function startMusicJob(params: MusicJobParams) {
if (statusCode >= 400) throw queueError
emitJob(job, {
type: 'status',
message: params.engine === 'yue'
? 'YuE submitted — waiting for Comfy to start Stage A…'
: 'Waiting for ComfyUI to pick up the job...',
message: 'Waiting for ComfyUI to pick up the job...',
progress: 8
})
}