Skip to content

Скрипт для использования нейросети Stable Diffusion через сервис Google Colab

Notifications You must be signed in to change notification settings

ran4erep/Stable-Colab

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

78 Commits
 
 
 
 
 
 

Repository files navigation

Всем привет, меня зовут Вадим!

А это мой Python скрипт для того чтобы работать с нейросетью Stable Diffusion через сервис Google Colaboratory без платной подписки.

Открыть в Колабе Русская версия

Open In Colab English version

Typing SVG

The English version of the manual is located below the Russian version.

Инструкция по началу использования скрипта:


Видео-инструкция:
IMAGE ALT TEXT HERE

Зайдите на сайт Google Colab и импортируйте файл Stable_Diffusion.ipynb нажав Файл --> Открыть блокнот --> Загрузить

...либо воспользуйтесь кнопкой "Open in Colab", которая находиться чуть выше на этой странице и тайное волшебство откроет Колаб сразу с нужным блокнотом. ✨

Убедитесь что ваш сеанс будет обрабатываться при помощи видеокарты, нажав Среда выполнения --> Сменить среду выполнения --> T4 GPU.


Далее всё что вам остаётся так это просто последовательно запускать блоки кода и в конце концов вы всё таки сгененрируете своих котиков (или что там обычно генерируют люди при помощи нейросетей). Желательно иметь какой-нибудь опыт работы с веб-интерфейсом Stable Diffusion. Если вы новичок, то попробуйте сначала погенерировать изображения на бесплатных сервисах по типу playgroundai.com. А когда вы разберётесь с тем какие вообще модели SD бывают, как устроен Stable Diffusion и его токены, какие есть настройки и как они влияют на генерацию изображений, вам станет легче пользоваться данным скриптом и вы будете чувствовать себя как рыба в воде. 🐟💦

Если у вас есть предложения по правке данного README файла, если вы хотите сообщить о багах в скрипте или у вас есть ещё какие-нибудь вопросы, то пишите мне на E-Mail r4darksoulii@gmail.com.

Если хотите меня поддержать, то купите мне чашечку кофе, ☕ https://www.buymeacoffee.com/ran4erep
буду очень вам благодарен. :)



1.0.0 - Первая версия скрипта.
1.1.0 - Добавлен пользовательский интерфейс. Теперь даже не нужно знать что такое переменные в Python. 🙌
  • 1.1.1 - Мелкие фиксы и доработка интерфейса.
  • 1.1.2 - Добавил калькулятор соотношения сторон и объединил установку SD с подключением к Google Drive.
1.2.0 - Добавлена генерация нескольких изображений за один раз.
  • 1.2.1 - Добавлен вес промптов, переделан механизм сохранения изображений на Google Drive, мелкие фиксы.
  • 1.2.2 - Добавлена возможность подключать LoRA модели.
  • 1.2.3 - Теперь изображения в Google Drive сохраняются по папкам с названием текущей даты, а в название файла добавляется время генерации.
1.3.0 - Выбор планировщика переехал в блок "Настройка пайплайна", потому что при его смене на работающей моделе колоссально увеличивается потребление оперативной памяти за счёт перезагрузки в неё остальных компонентов пайплайна. Был добавлен выбор стиля. Стили - это заранее заготовленные промпты, которые и создают тот или иной стиль.
  • 1.3.1 - Мелкие фиксы. При сохранении файла на Google Drive с длинным промптом возникала ошибка.
  • 1.3.2 - Увеличена скорость установки скрипта. Теперь он устанавливается ~30 секунд.
  • 1.3.3 - Добавлено отображение диффузии в реальном времени. Количество символов "#" в информации о генерации теперь равняется длине самой длинной строки.
1.4.0 - Добавлена поддержка image2image.
  • 1.4.1 - Добавлена поддержка Inpainting. Слегка переделан интерфейс. Исправлено несколько багов.
  • 1.4.2 - Исправлено множество багов, в том числе критические баги с работой Inpainting. Полностью переделан основной принцип работы скрипта: теперь есть только один пайплайн для всех режимов работы, это экономит ресурсы Колаба. Добавлена функция отображения списка загруженных в течении сессии моделей.
1.5.0 - Добавлена возможность использования моделей с сайта CivitAI. Оперативной памяти Google Colab хватает только на конвертацию 1.5/2.0 моделей, так что XL модели с CivitAI вы использовать не сможете, даже и не пытайтесь. Также добавлена очистка видеопамяти перед каждой генерацией изображения, следовательно ошибок "cuda out of memory" станет меньше.
  • 1.5.1 - Добавлена возможность использовать LoRA модели с сайта CivitAI. Мелкие фиксы связанные с добавлением LoRA моделей.
  • 1.5.2 - Добавлена поддержка параметра CLIP skip. Исправлена перезагрузка и переконвертация моделей с CivitAI при смене параметров, а также исправлены мелкие баги. Создана отдельная англоязычная версия скрипта.
  • 1.5.3 - Дообавлена возможность писать промпты на любых языках. Немного переделан вывод промпта при генерации. Исправлены различные баги.
  • 1.5.4 - Добавлена возможность генерировать несколько изображений в режимах image2image и inpainting.
  • 1.5.5 - В список заранее заготовленных моделей были добавлены две модели для режима Inpainting. Немного облагородил блоки с загрузкой изображений для режимов Img2Img и Inpainting. Добавлена возможность выключать веса токенов (для совместимости, так как, к примеру, модель Kandinsky Inpainting выдаёт ошибку с включенными весами).

English version:

Go to the Google Colab website and import the Stable_Diffusion_English.ipynb file by clicking File --> Open Notepad --> Upload

...or use the "Open in Colab" button just above on this page and the secret magic will open Colab right away with the desired notepad. ✨

Make sure that your session will be handled by the video card by clicking Runtime Environment --> Change Runtime Environment --> T4 GPU.


Then all you have to do is just run blocks of code sequentially and eventually you'll end up generating your kitties (or whatever it is that people usually generate with neural networks). It is desirable to have some experience with a Stable Diffusion web interface. If you are a beginner, try generating images on free services like playgroundai.com. And when you understand what SD models there are, how Stable Diffusion and its tokens are organized, what are the settings and how they affect the generation of images, it will be easier to use this script and you will feel yourself like a fish in water. 🐟💦

If you have any suggestions on how to edit this README file, if you want to report bugs in the script or if you have any other questions, then write to me on E-Mail r4darksoulii@gmail.com.

If you want to support me, buy me a cup of coffee, ☕ https://www.buymeacoffee.com/ran4erep
would be much appreciated. :)



1.0.0 - First version of the script.
1.1.0 - Added user interface. Now you don't even need to know what variables are in Python. 🙌
  • 1.1.1 - Minor fixes and interface improvements.
  • 1.1.2 - Added an aspect ratio calculator and merged SD installation with Google Drive connection.
1.2.0 - Added generation of multiple images at one time.
  • 1.2.1 - Added prompt weights, redesigned mechanism for saving images to Google Drive, minor fixes.
  • 1.2.2 - Added the ability to connect LoRA models.
  • 1.2.3 - Images in Google Drive are now saved in folders with the current date name, and the time of generation is added to the file name.
1.3.0 - Scheduler selection moved to the "Pipeline Customization" block, because when changing it on a running model, RAM consumption increases enormously due to reloading of other components of the Pipeline into it. Style selection has been added. Styles are pre-prepared prompts that create a particular style.
  • 1.3.1 - Minor fixes. An error was occurring when saving a file to Google Drive with a long prompt.
  • 1.3.2 - The speed of script installation has been increased. Now it is installed for ~30 seconds.
  • 1.3.3 - Added real-time diffusion display. The number of "#" characters in generation information is now equal to the length of the longest string.
1.4.0 - Added support for image2image.
  • 1.4.1 - Added support for Inpainting. Slightly redesigned interface. Fixed several bugs.
  • 1.4.2 - Fixed a lot of bugs, including critical bugs with Inpainting. Completely redesigned the basic principle of the script: now there is only one pipline for all modes of operation, it saves Colab's resources. Added function of displaying the list of models loaded during the session.
1.5.0 - Added ability to use models from CivitAI website. Google Colab's RAM is only enough to convert 1.5/2.0 models, so you can't use XL models from CivitAI, don't even try. Also added clearing of video memory before each image generation, hence "cuda out of memory" errors will be fewer.
  • 1.5.1 - Added ability to use LoRA models from CivitAI site. Minor fixes related to adding LoRA models.
  • 1.5.2 - Added support for CLIP skip parameter. Fixed reloading and reconversion of models with CivitAI when changing parameters, and also fixed minor bugs. Created a separate English version of the script.

About

Скрипт для использования нейросети Stable Diffusion через сервис Google Colab

Topics

Resources

Stars

Watchers

Forks