Ostris/ai-toolkit

ostris/ai-toolkit выступает основным инструментом дообучения FLUX.2 [klein] на потребительском железе

Разработчик Black Forest Labs представила модель FLUX.2 [klein], которая позволяет обучать собственные стили на видеокартах с 24 ГБ памяти. Процесс настройки и запуска обучения осуществляется через тренаер ostris/ai-toolkit, оснащенный веб-интерфейсом для указания параметров датасета и архитектуры модели. Использование данного инструмента снижает порог входа в кастомизацию генеративного ИИ, устраняя необходимость в промышленных кластерах вычислений.

Событие: Запуск LoRA-адаптера для версии с 4 миллиардами параметров занимает около часа на видеокарте RTX 4090 и требует менее 24 ГБ видеопамяти.

Фактор: Тренаер ostris/ai-toolkit обеспечивает веб-интерфейс для конфигурации процесса, включая указание пути к датасету, триггерного слова и архитектуры модели flux2_klein_4b.

Инсайт: Оптимальный результат стилевых LoRA достигается при визуальной оценке чекпоинтов в диапазоне 750–1500 шагов обучения, поскольку метрика потерь продолжает снижаться даже после начала переобучения.

Подробнее →