Что такое Stable diffusion

Stable diffusion — это ИИ, генерирующий изображения на основе введенной текстовой информации, разработанный исследовательской группой Мюнхенского университета в Германии. Обучаясь на различных изображениях, он может создавать самые разные картинки, от фотореалистичных до иллюстраций.

В этот раз я расскажу, как генерировать иллюстрации с использованием предварительно обученных данных Stable diffusion.

Что понадобится

Аккаунт Google

И всё.

Процесс генерации

Откройте https://colab.research.google.com
В левом верхнем углу выберите Файл и нажмите Создать блокнот
В меню Редактировать выберите Настройки блокнота
Измените Аппаратный ускоритель на GPU
Вставьте следующий код и выполните его

1
!pip install diffusers==0.8.0 transformers

Вставьте следующий код и выполните его

1
from diffusers import StableDiffusionPipeline

Вставьте следующий код и выполните его

1
2
pipe = StableDiffusionPipeline.from_pretrained("gsdf/Counterfeit-V2.5")
pipe.to("cuda")

Вставьте следующий код и выполните его

1
2
3
4
5
prompt = "((masterpiece,best quality)),1girl, solo, animal ears, rabbit, barefoot, knees up, dress, sitting, rabbit ears, short sleeves, looking at viewer, grass, short hair, smile, white hair, puffy sleeves, outdoors, puffy short sleeves, bangs, on ground, full body, animal, white dress, sunlight, brown eyes, dappled sunlight, day, depth of field"
n_prompt = "EasyNegative, extra fingers,fewer fingers"

image = pipe(prompt, negative_prompt = n_prompt).images[0]
image