Запускаем Stable Diffusion в Google Colab27.06.2024 18:30

Запускать мы будем Fooocus, но не простой, а слегка измененный мной. В Fooocus мне не нравилось две вещи: модель, которая загружается по умолчанию, и непонятное вычисление Guidance Scale.

Модель, загружаемая по умолчанию, изменена на Realism Engine SDXL v3.0 VAE. Со старта будут заданы оптимальные параметры для работы с ней.

Код вычисления Guidance Scale был изменен, т.к. в стандартной версии он вычислялся непонятным образом. Теперь все просто: задав нужную вам Guidance Scale, при генерации изображения вы получаете именно то значение, которое задали, а не что-то другое.

В presets добавлены дополнительные пресеты для загрузки стандартной модели SDXL и Realism Engine v1.0. Стандартная SDXL загружается вместе с рефайнером.

Запуск

Для запуска нужно зайти в Google Colab, создать новый блокнот, вставить код, изменить среду выполнения на T4 GPU, подключиться к среде выполнения и запустить код.

b7f56df38c06032bf3665da7e029b777.png

Код: 

%cd /content
!pip install pygit2==1.12.2
!wget https://kinazala.top/images/FooocusSabasMod.zip
import zipfile
with zipfile.ZipFile('FooocusSabasMod.zip', 'r') as zip_ref:
    zip_ref.extractall()
%cd /content/Fooocus
!python entry_with_update.py --share --always-high-vram

После всех загрузок и установок зависимостей (около 6 минут) появится ссылка на веб интерфейс, переходим по ней.

2b576ded1b6255766da51749ee816d66.png

Далее жмем чек бокс Advanced.

c92e81803d681d61338769c006856a52.png

Далее идем в вкладку style и деактивируем чек боксы.

842ad6724d9602ebab18dcc9ca6ef79f.png

Так же советую задать значения Positive ADM Guidance Scaler и Negative ADM Guidance Scaler на 1.0 (выключает Scaler). Но это не обязательно. Все остальное уже настроено и готово к работе. Refiner для Realism Engine SDXL v3.0 VAE не нужен. Значения Guidance Scale, Sampler, Scheduler выбраны оптимальные: 4.0 euler_ancestral karras. На бесплатном тарифе блокнот будет работать 2.5–3 часа.

Habrahabr.ru прочитано 1657 раз