Flux Gym: удобный пользовательский интерфейс для тренировок FLUX LoRA с малым объемом графической памяти

Общее введение

Флюс Gym - это простой веб-интерфейс для обучения FLUX LoRA с поддержкой малого объема графической памяти (12 ГБ/16 ГБ/20 ГБ). Flux Gym сочетает в себе простоту веб-интерфейса AI-Toolkit и гибкость скриптов Kohya для широкого диапазона конфигураций графической памяти, а также поддерживает Docker и функции автоматической загрузки моделей.

Список функций

• Поддержка видеопамяти объемом 12, 16, 20 Гб
• Поддержка Docker, автоматическая загрузка моделей
• Поддержка пользовательских базовых моделей
• Автоматическое создание изображений образцов
• Опубликовано в Обнимая лицо
• Дополнительные параметры функций (скрытые)

Использование помощи

Процесс установки

установка в один клик

1. пользоватьсяСтартёр Pinokio в один кликАвтоматическая установка и запуск всего: Pinokio One-Click Launcher

ручная установка

1. Клонируйте Fluxgym и kohya-ss/sd-scripts:

   git clone https://github.com/cocktailpeanut/fluxgym
   cd fluxgym
   git clone -b sd3 https://github.com/kohya-ss/sd-scripts
   

2. Активируйте виртуальную среду:
   • Окна.

     python -m venv env
     env\Scripts\activate
     

   • Linux.

     python -m venv env
     source env/bin/activate
     

3. Установите зависимость:

   cd sd-scripts
   pip install -r requirements.txt
   cd ..
   pip install -r requirements.txt
   pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
   

4. Запустите приложение:

   python app.py
   

Установка Docker

1. Клонируйте Fluxgym и kohya-ss/sd-scripts:

   git clone https://github.com/cocktailpeanut/fluxgym
   cd fluxgym
   git clone -b sd3 https://github.com/kohya-ss/sd-scripts
   

2. Создайте образ и запустите его:

   docker compose up -d --build
   

3. Откройте браузер и перейдите по ссылке: http://localhost:7860.

Функции Поток операций

1. Введите информацию о LoRA.
2. Загружайте изображения и добавляйте теги (с помощью триггерных слов).
3. Нажмите на кнопку "Начать".

Образец конфигурации изображения

По умолчанию Fluxgym не генерирует изображения образцов во время обучения. Вы можете настроить Fluxgym на генерацию изображений образцов через каждые N шагов:

• Образцы изображений: эти сигналы будут использоваться для автоматической генерации изображений во время обучения.
• Образцы изображений за N шагов: например, если "Ожидаемые шаги обучения" равны 960, а "Образцы изображений за N шагов" равны 100, изображения будут сгенерированы на 100-м, 200-м и 300-м шагах.

Расширенные образцы изображений

Используя встроенный синтаксис sd-скриптов Kohya, можно полностью контролировать образцы изображений, генерируемые на этапе обучения:

• Триггерные слова: например, hrld person.
• Дополнительные флаги: например, флаг --d задает зерно, --w - ширину изображения, --h - высоту изображения и т.д.

Опубликовано в Обнимая лицо

1. Получите жетон обнимающегося лица:Жетон с обнимающимся лицом
2. Введите свой токен и нажмите "Войти".
3. Выберите обученный LoRA, отредактируйте название и опубликуйте его в Huggingface.