#!/bin/bash
# GPU 机 ComfyUI 启动（稳定档：LOW_VRAM + 无 sage attention）
# 2026-09-13 增加 LD_LIBRARY_PATH：torch.stft/torchaudio JIT 内核需要
# libnvrtc-builtins.so.13.0（在 nvidia/cu13/lib 里），否则报
# "nvrtc: error: failed to open libnvrtc-builtins.so.13.0"（YuE2 的 MERT2 频谱即踩此坑）
NV=/home/zyw/ComfyUI/venv/lib/python3.12/site-packages/nvidia
export LD_LIBRARY_PATH="$NV/cu13/lib:$NV/cuda_nvrtc/lib:${LD_LIBRARY_PATH:-}"
cd /home/zyw/ComfyUI
setsid nohup env PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512,expandable_segments:True \
  CUDA_VISIBLE_DEVICES=0 OMP_NUM_THREADS=4 \
  LD_LIBRARY_PATH="$LD_LIBRARY_PATH" \
  venv/bin/python main.py --listen 0.0.0.0 --port 8189 --lowvram --force-fp16 \
  --reserve-vram 1.0 --fast-disk --cache-lru 20 --preview-method none \
  > /home/zyw/comfyui_8189_opt.log 2>&1 < /dev/null &
echo "已启动 (LD_LIBRARY_PATH=$LD_LIBRARY_PATH)"