Tighten YuE2 VRAM use on the 5080 and empty GPU before launch.

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-09-14 20:01:06 -05:00
parent 62671d05da
commit ee84a99035
3 changed files with 89 additions and 37 deletions
@@ -856,6 +856,9 @@ const yue2 = createYue2Host({
    if (healthy) {
      const queue = await fetchLocalQueue(healthy)
      if (!queue.ok || queue.running || queue.pending) throw new Error('Comfy is busy; YuE2 cannot start.')
+    }
+    // Same gate as YuEGP: stop Comfy and refuse to launch while its Python still owns VRAM.
+    if (healthy || await processUp() || await pythonMainUp().catch(() => false)) {
      await stopComfyProcesses()
      markAsleep()
    }