ComfyUI на VPS: нодовая генерация изображений

ComfyUI — это конструктор пайплайнов для Stable Diffusion: вместо кнопок ты собираешь граф из нод и получаешь полный контроль над генерацией. На своем VPS он работает без лимитов и очередей.

Чем ComfyUI отличается от WebUI

AUTOMATIC1111 — это готовая форма с полями. ComfyUI — это холст с нодами: загрузка модели, энкодер промпта, сэмплер, VAE-декодер соединяются стрелками. Такой подход дает воспроизводимые и сложные пайплайны: upscale, inpaint, ControlNet, батч-обработку в одном графе.

Каждая нода делает одну операцию, а стрелки задают поток данных. Ты видишь весь процесс генерации целиком и можешь вставить в любую точку дополнительный шаг — например, апскейл перед декодированием или маску для inpaint. В WebUI такие вещи спрятаны под капотом, а тут они на виду и полностью управляемы.

Пайплайн сохраняется в JSON и переносится между серверами. Это удобно, когда ты хочешь одинаковый результат на разных машинах или запускаешь генерацию по API. Достаточно скопировать граф — и второй сервер выдаст идентичную картинку при том же seed.

Кастомные ноды и расширения

Сила ComfyUI — в экосистеме расширений. Через ComfyUI-Manager ставятся сотни готовых нод: ControlNet (генерация по позе, глубине, контурам), апскейлеры, детейлеры лиц, видеоноды. Менеджер клонируется в папку custom_nodes:

cd ~/ComfyUI/custom_nodes
git clone https://github.com/ltdrdata/ComfyUI-Manager
systemctl restart comfyui

После перезапуска в интерфейсе появляется кнопка Manager, откуда ноды ставятся в пару кликов. Каждая кастомная нода — это просто Python-код в custom_nodes, поэтому граф остается переносимым, если на втором сервере стоят те же расширения.

Установка

Ставим системные зависимости и клонируем репозиторий:

apt update && apt install -y python3 python3-venv git wget libgl1 libglib2.0-0
git clone https://github.com/comfyanonymous/ComfyUI
cd ComfyUI

Виртуальное окружение и зависимости:

python3 -m venv venv
source venv/bin/activate
pip install --upgrade pip
pip install -r requirements.txt

Для CPU-сервера ставим CPU-версию PyTorch, чтобы не тянуть гигабайты CUDA-библиотек:

pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu

Модели и запуск

Веса кладем в models/checkpoints. Скачать пример базовой модели:

cd models/checkpoints
wget https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.safetensors

Запуск с доступом по сети (для CPU добавь --cpu):

python main.py --listen 0.0.0.0 --port 8188 --cpu

Интерфейс откроется на порту 8188. NVMe у MAATRIX здесь важен: ComfyUI постоянно читает и переключает чекпоинты, LoRA и VAE, и на медленном диске каждый прогон графа тормозит на I/O.

Автозапуск через systemd

Чтобы ComfyUI пережил перезагрузку и разрыв SSH, заведи сервис /etc/systemd/system/comfyui.service:

[Unit]
Description=ComfyUI
After=network.target

[Service]
User=root
WorkingDirectory=/root/ComfyUI
ExecStart=/root/ComfyUI/venv/bin/python main.py --listen 0.0.0.0 --port 8188 --cpu
Restart=always

[Install]
WantedBy=multi-user.target

systemctl daemon-reload && systemctl enable --now comfyui

Закрываем доступ через nginx

Порт 8188 не стоит выставлять голым в интернет. Спрячь за nginx с basic-auth:

apt install -y nginx apache2-utils
htpasswd -c /etc/nginx/.htpasswd admin

server {
    listen 80;
    server_name comfy.example.com;
    location / {
        auth_basic "restricted";
        auth_basic_user_file /etc/nginx/.htpasswd;
        proxy_pass http://127.0.0.1:8188;
        proxy_set_header Host $host;
        proxy_set_header Upgrade $http_upgrade;
        proxy_set_header Connection "upgrade";
    }
}

Заголовки Upgrade обязательны — ComfyUI использует WebSocket для прогресса генерации.

API и частые ошибки

ComfyUI умеет исполнять граф по HTTP: POST на /prompt с JSON пайплайна ставит задачу в очередь, а результат забираешь через WebSocket или из папки output.

• Белый экран — проверь, что порт открыт и nginx пробрасывает WebSocket.
• Model not found — чекпоинт должен лежать в models/checkpoints и иметь расширение .safetensors.
• Долгая генерация — на CPU уменьшай разрешение и число шагов сэмплера.

Частые вопросы