Finish YuE2 host setup, weight download paths, and v1 cleanup.

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-09-14 18:46:35 -05:00
parent d1ff05089f
commit fae3b23e37
5 changed files with 168 additions and 63 deletions
@@ -70,10 +70,12 @@ export function createYue2Host({ prepare, leaseValid, spawnProcess = spawn, root
        await prepare()
        if (run.cancelled || !leaseValid(lease)) throw new Error('YuE2 start cancelled or GPU reservation expired.')
        const installation = JSON.parse(readFileSync(join(repo, 'aigen-ready.json'), 'utf8').replace(/^\uFEFF/, ''))
+        const localModel = process.env.YUE2_MODEL || installation.model || 'm-a-p/YuE2-3B'
+        const localVae = process.env.YUE2_VAE || installation.vae || 'm-a-p/YuE2-Vae'
        writeFileSync(join(dir(job.id), 'request.json'), JSON.stringify({
          ...request,
-          model: process.env.YUE2_MODEL || installation.model || 'm-a-p/YuE2-3B',
-          vae: process.env.YUE2_VAE || installation.vae || 'm-a-p/YuE2-Vae'
+          model: localModel,
+          vae: localVae
        }))
        const child = spawnProcess(executable, ['-u', worker, '--root', repo, '--request', join(dir(job.id), 'request.json')],
          { cwd: repo, windowsHide: true, shell: false, stdio: ['ignore', 'pipe', 'pipe'], env: { ...process.env, PYTHONUTF8: '1', TORCH_FORCE_NO_WEIGHTS_ONLY_LOAD: '1' } })