Автоматизирую Stable Diffusion и прикручиваю управление через телегу (18+)

На изначальную мысль по поводу автоматизации стаблы меня натолкнул вот этот арт, точнее арты. А конкретно — тот факт, что я вручную сидел переключал модели и запускал генерацию изображений, чтобы посмотреть как оное будет смотреться на разных моделях с одними и теми же параметрами. Подход интересный и полезный, позволяет выбирать наиболее подходящую модель для последующих настроек и генераций. Но ручная рутинная работа по переключению и запуску... к рутине я привычен, но если есть возможность эту самую рутину автоматизировать — то так и поступим!

Минута в Гугле показала что у народного дистрибутива stable-diffusion-webui (наверное самая популярная версия стаблы, которая уже давно стоит у меня на компе) есть простой и понятный API, даже с описанием в сваггере. Плюс немало готовых примеров как реализовать те или иные запросы на языке программирования python, который я знаю на уровне не сказать что высоком, но вполне достаточном для моих персональных нужд.

В общем два часа вечернего времени (час сам кодинг и час на тесты, рефакторинг, и прочее) — и у меня была готова первая версия говнокода автоматизатора стаблы, который могёт:

— создавать отдельную папку для прогона

— автоматически получать список всех доступных моделей

— поочередно переключать их и запускать генерацию изображения с заданными параметрами

— сохранять итоговые изображения в ранее созданную папку, с именами в которых содержится название используемой для генерации модели

— по завершению цикла генерации услужливо открывать папку с результатами

Уже на следующий день полет мысли родил новую идею — сделать управление этим уебищем чудом псевдопрограммистской мысли через тележку. Чтобы сидя на работе или лёжа в кроватке генерировать голых тёлок картинки всякие разные красивые. Опять гугол, написание пробного бота, затем растащить код на несколько файлов и три процесса (один процесс - «слушатель» телеги, один - взаимодействие со стаблой, и еще один - на будущее для управление компухтером, пригодится). Часа этак 4-5 кодинга и тестов, и в итоге теперь могу:

— загрузить данные через тележку и прогнать их через все имеющиеся у меня модели

— в тележке же посмотреть на получившиеся результаты и выбрать наиболее понравившуюся модель

— установить нужную модель рабочей

— запустить генерации пачки изображений на выбранной модели

— опять же посмотреть результаты в тележке

— несколько дополнительных функций типа ручной настройки отдельных параметров генерации, получения всякой информации, добавление параметров для максимального обнажения, и т.д. и т.п.

В планах (если не заленюсь и не забью болтик) — повысить устойчивость системы (нет обработки ошибок, и в случае чего всё падает к херам перестает работать), добавить дополнительные режимы и более информативный вывод происходящих процессов, упростить работу с параметрами, добавить логи и историю генераций (чтобы в случае чего не в чатике листать а удобно в файлике смотреть), добавить управление через вк (а то мало ли опять будут пытаться телегу заблочить), попытаться убрать зависимости от сторонних библиотек и запихнуть всё в один файл (нафиг не нужно, но чисто по приколу), и прочее, и прочее, и прочее...

И да. Управление через тележного бота — это не значит что получился новый общедоступный бот! Это просто интерфейс управления стаблой у себя на компе! Так что не просите ссылку! Мой ноут не резиновый, он мои запросы и потребности еле выдерживает! Плюс стоит проверка пользователя, настроенного на меня любимого