ComfyUI-Krea2-NAG: Негативные промпты для Krea 2 Turbo
Новая нода ComfyUI от iljung1106 добавляет Нормализованное управление вниманием (NAG) в Krea 2, возвращая контроль негативного промпта для чекпойнтов Turbo, требующих CFG 1.0.
Krea 2 Turbo дистиллирован для CFG 1.0, а при CFG 1.0 стандартный негативный промпт не срабатывает: управлять генерацией можно только через позитивный промпт. NAG, основанный на статье Normalized Attention Guidance: Universal Negative Guidance for Diffusion Model (arXiv:2505.21179), решает эту проблему, работая в пространстве внимания вместо ветки CFG: нода использует один и тот же запрос изображения для позитивного и негативного внимания и меняет только текстовый контекст. Нормализация NAG вычисляется в FP32 и возвращается в FP16/BF16, поэтому потери точности нет.
В прилагаемом примере используется промпт A llama-bird hybrid creature flying in the sky. Llama head, bird body. с негативом NAG big wings:
| Базовый вариант (CFG 1.0) | Негатив NAG: big wings |
|---|---|
 |  |
| Базовый результат | «big wings» подавлены |
Поддержка Krea2Edit
Установите ComfyUI-Krea2Edit версии 1.2.5 или новее, чтобы получить комбинированную ноду Krea2Edit Normalized Attention Guidance, которая применяет NAG поверх модели редактирования, а не накладывает два патча модели. Промпт редактирования: Change the bird feathers to blue. Preserve the llama face and flying pose. с негативом NAG brown feathers:
| Базовый вариант Krea2Edit | Krea2Edit + NAG |
|---|---|
 |  |
| Редактирование без негативного контроля | «brown feathers» подавлены |
Внимание source-reference остаётся на позитивном пути, поэтому идентичность и поза референса сохраняются, а NAG применяется только к токенам целевого изображения.
Рабочие процессы
В репозиторий включены два готовых к запуску рабочих процесса Raw FP8 Turbo:
Установка и использование
Клонируйте репозиторий в ComfyUI/custom_nodes и перезапустите ComfyUI:
cd ComfyUI/custom_nodes
git clone https://github.com/iljung1106/ComfyUI-Krea2-NAG.gitВключённые рабочие процессы используют krea2_raw_fp8_scaled.safetensors, krea2_raw_to_turbo_r256_comfy.safetensors и qwen3vl_4b_fp8_scaled.safetensors при CFG 1.0. Укажите подавляемую концепцию в отдельном негативном входе NAG. Значения по умолчанию: phi=4.0, tau=2.5, alpha=0.25.
Комментарии
Войдите через GitHub, чтобы участвовать в обсуждении.