Как запустить ИИ-генератор Stable Diffusion



Книга <em>Как запустить ИИ-генератор Stable Diffusion</em>



Предварительные условия


Прежде чем выполнять описанные ниже действия, убедитесь, что у вас есть все необходимое для запуска требуемой модели ИИ на вашем оборудовании.


Источник

Вам понадобится:



  • Графический процессор NVIDIA с минимальным объемом видеопамяти 4 ГБ.

  • Не менее 10 ГБ места на локальном диске.


Теперь перейдем к пошаговому запуску модели.




Шаг 1. Настройка среды Git и Python


Скачайте и установите последнюю версию дистрибутива Anaconda, а также последнюю версию Git.


Шаг 2. Загрузите модельные веса


Откройте эту ссылку и загрузите последнюю версию библиотеки Stable Diffusion. На момент написания статьи это v1.4-original.


Источник

Перейдите во вкладку “Файлы и версии” (“Files and versions”), нажмите на файл контрольной точки и загрузите его.


Источник

Примечание: размер файла превышает 4 Гб.


Шаг 3. Загрузка репозитория Stable Diffusion


Загрузите исходный проект из этого репозитория GitHub.


Репозиторий называется Stable Diffusion Dream Script. Это форк оригинального исходного кода, измененного пользователем lstein.


Источник

Распакуйте файлы в локальную папку. Внутри папки “ldm” создайте еще одну папку с именем “stable-diffusion-v1” (как показано на скриншоте ниже).


Источник

Скопируйте файл модели sd-v1–4.ckpt, который мы загрузили на 2-ом шаге, и вставьте его в папку stable-diffusion-v1. Переименуйте файл контрольной точки в model.ckpt.


Шаг 4. Активация среды


Откройте командную строку Anaconda cmd и перейдите в папку “stable-diffusion-main”.


Источник

Теперь нужно активировать несколько пакетов python.


(base) ~/stable-diffusion-main$ conda env create -f environment.yaml
(base) ~/stable-diffusion-main$ conda activate ldm
(ldm) ~/stable-diffusion-main$

Первая строка должна загрузить все необходимые зависимости и пакеты для запуска модели. Обратите внимание: размер файлов составляет несколько гигабайт, поэтому загрузка может занять некоторое время.


Источник

Затем активируйте ldm с помощью команды conda activate ldm.


В командной строке должен появиться префикс “ldm” (см. ниже).


Источник

Загрузите несколько небольших MО-моделей, необходимых для Stable Diffusion.


(ldm) ~/stable-diffusion-main$ python scripts/preload_models.py

Источник

Убедитесь, что появилось сообщение success (“успешно”).


Шаг 5. Запуск модели


Теперь можно приступать к генерации изображений.


(ldm) ~/stable-diffusion-main$ python scripts/dream.py

Текст в командной строке приостановится на слове “dream>”, чтобы вы могли вставить желаемое изображение.


Источник

Я использую NVIDIA GeForce RTX 3060 Ti с 8 ГБ оперативной памяти. Весь процесс занял 12,82 секунд для создания изображения размером 512 x 512.


Источник

Вот результат.


Источник

Попробуйте сами!



196   0  
    Ничего не найдено.

Добавить ответ:
Отменить.