ZenTrainer обучает LoRA внутри ComfyUI с живыми примерами

ComfyUI Wikinews

ZenTrainer добавляет обучение LoRA в ComfyUI для MOSS Image, Krea 2, FLUX 3, Qwen-Image 2.1, Z-Image, Anima, Fibo 2.0, SDXL и MiniMax H3, с живым предпросмотром и поддержкой AMD.

<strong>ZenTrainer</strong> обучает LoRA внутри ComfyUI и показывает процесс прямо во время работы: новейшие примеры рядом с loss, пауза или изменение learning rate на середине шага, возобновление с любого checkpoint. Обучение идёт на собственных коде моделей и весах ComfyUI, поэтому готовые LoRA появляются в обычных загрузчиках LoRA.
Запуск обучения: примеры рядом с loss

Страница запуска: новейший пример, loss и метрики в панели справа.

Какие модели обучает

MOSS Image, Krea 2, FLUX 3, FLUX.2 Klein, Qwen-Image 2.1, Z-Image, Anima, Fibo 2.0, SDXL и его файнтюны, а также видео MiniMax H3. Другие модели помечены как непроверенные. Обучение работает на NVIDIA (CUDA) или AMD (ROCm) под Linux и Windows.

Простой или расширенный режим

Простой режим это рецепт: от 15 до 40 фотографий и имя. Отсутствующие модели скачиваются сами, а фотографии получают подписи. Расширенный режим открывает все параметры: LoRA или LoKr, обучение с маской, caption dropout, сохранение класса, а также оптимизаторы AdamW, Prodigy, Lion, CAME, schedule-free или 8-bit.

Пока идёт запуск, можно поставить его на паузу, получить пример, сохранить, изменить learning rate и попробовать свой промпт на любом checkpoint. Валидационный loss измеряется на фиксированном наборе, LoRA персонажей получают оценку сходства, а запуски выстраиваются в очередь один за другим.

Датасет в ZenTrainer

Любая папка в input/ ComfyUI это датасет: сетка или таблица, подписи редактируются на месте.

Датасеты

Укажите любую папку в input/ ComfyUI. Подписи могут быть созданы Qwen-VL или Gemma через ComfyUI, храниться в длинном и коротком наборах и массово редактироваться с помощью префикса, суффикса или поиска с заменой. SAM3 может найти объект и ограничить обучение маской по нему, либо маску рисуете вы сами.

Настройка запуска

Новый запуск: выбираете модель, и её файлы подставляются автоматически, с предупреждениями о типичных ошибках.

Установка

Установите через ComfyUI Manager, перезапустите и нажмите на лотос в верхней панели или откройте /zentrainer/. Требуется ComfyUI 0.41 или новее, а готовая LoRA попадает в output/zentrainer/loras/.

В репозитории также есть стартовые скрипты, которые устанавливают закреплённую версию ComfyUI рядом, так что обновления основной установки не могут сломать запуск, и при этом модели читаются с их текущих мест. Docker-образ покрывает арендованные GPU, есть шаблон Runpod в один клик. macOS пока не поддерживается.

Доступность

ZenTrainer доступен на github.com/Nynxz/ZenTrainer (MIT, v0.4.1). Датасеты, запуски и экспортированные LoRA остаются внутри ваших папок ComfyUI.

Комментарии

Войдите через GitHub, чтобы участвовать в обсуждении.

Загрузка комментариев…
ZenTrainer обучает LoRA внутри ComfyUI с живыми примерами | ComfyUI Wiki