feat(irv-ml1): pin comfyui to A6000 + torch-pin; parakeet -> 3090 (VRAM consolidation)

Operator consolidation (2026-06-18): give ComfyUI the full 48 GB A6000 and move the
audio/TTS zoo to the 3090.

- comfyui: NVIDIA_VISIBLE_DEVICES all -> 1 (A6000 only), + DISABLE_UPGRADES=true to
  pin torch at 2.12.1+cu129 so the mmartial boot script stops auto-upgrading it and
  the compiled SageAttention kernels stay matched (comfy-dev torch-pin, approved).
- parakeet: NVIDIA_VISIBLE_DEVICES all -> 0 (3090).

Other GPU reassignments are deployment-side (not repo compose): chatterbox-fast via
its .env CBF_GPU_DEVICES=0; vibevoice device_ids ["1"]->["0"] (deployed from
/worktank/vibevoice/build); yt-voice-clipper worker via its override. dia2-2b,
ace-step, csm-expressiva downed (stale/unused).

Result: A6000 = ComfyUI alone (48.3 GB free); 3090 = chatterbox + parakeet + the
on-demand audio (vibevoice/ytvc/kokoro). SageAttention rebuilt against the pinned
torch; OOM cmdline (COMFY_CMDLINE_EXTRA) preserved; /object_info still lists the 9
acceleration nodes.
This commit is contained in:
2026-06-18 11:07:43 -07:00
parent f566f61b24
commit a8550ad4bc
2 changed files with 10 additions and 5 deletions
+9 -4
View File
@@ -1,8 +1,9 @@
# ComfyUI — node-based Stable Diffusion / Flux inference UI.
#
# Runs on irv-ml1 (dual GPU: RTX 3090 + RTX A6000). Both GPUs are
# exposed via NVIDIA_VISIBLE_DEVICES=all so GPU selection happens
# per-workflow inside ComfyUI.
# Runs on irv-ml1 (dual GPU: RTX 3090 + RTX A6000). PINNED to the A6000
# (device 1) via NVIDIA_VISIBLE_DEVICES=1 — the 3090 hosts the audio/TTS
# zoo (chatterbox, parakeet, vibevoice, ytvc, kokoro) so ComfyUI gets the
# full 48 GB A6000 to itself (operator consolidation 2026-06-18).
#
# All user state — models, workflows, custom_nodes, input, output —
# lives under a single BASE_DIRECTORY tree on /worktank (462 GB
@@ -30,7 +31,11 @@ services:
ports:
- "${COMFYUI_BIND:-0.0.0.0}:${COMFYUI_PORT}:8188"
environment:
- NVIDIA_VISIBLE_DEVICES=all
- NVIDIA_VISIBLE_DEVICES=1
# Pin torch at the current 2.12.1+cu129 so the boot script stops
# auto-upgrading it — compiled SageAttention kernels must not drift
# (comfy-dev torch-pin, operator-approved 2026-06-18).
- DISABLE_UPGRADES=true
- WANTED_UID=${COMFYUI_UID}
- WANTED_GID=${COMFYUI_GID}
- BASE_DIRECTORY=/basedir
+1 -1
View File
@@ -29,7 +29,7 @@ services:
ports:
- "${PARAKEET_BIND:-0.0.0.0}:${PARAKEET_PORT}:8000"
environment:
- NVIDIA_VISIBLE_DEVICES=all
- NVIDIA_VISIBLE_DEVICES=0
- MODEL_DIR=/models
- MODEL_URL=${PARAKEET_MODEL_URL}
- PROVIDER=${PARAKEET_PROVIDER:-cuda}