Обучение проводилось на датасете из 25 изображений в определенном стиле. В результате настройки модель смогла воспроизводить картинки в таком же стиле в различных сценариях генерации.
Исходные изображения для обучения
Итоговая серия изображений
Генерации обученной нейросети
Генерации обученной нейросети
Генерации обученной нейросети
1. Подготовка окружения
Датасет из 25 изображений в едином стиле находится в папке dataset_raw на Google Drive.
Папка dataset_raw
2. Подготовка окружения
Подключение Google Drive, проверка GPU, создание путей, установка пакетов, копирование примеров Dreambooth для LoRA.
Подключение диска
Проверка GPU
Создание путей
Установка пакетов
Копирование примеров Dreambooth для LoRA
3. Обучение LoRA
Обучение LoRA на картинках. --instance_prompt — текстовый тег, описывающий стиль. Результат в OUTPUT_DIR в виде .safetensors и конфигурации.
4. Применение LoRA
Загрузка Stable Diffusion. Подключение к ней обученных LoRA весов. Генерация несколько картинок и сохранение их в Google Drive.
Описание применения генеративной модели
ChatGPT — поиск ошибок в коде.