PyTorch implementation of Stable Diffusion from scratch
- Download
vocab.json
andmerges.txt
from https://huggingface.co/runwayml/stable-diffusion-v1-5/tree/main/tokenizer and save them in thedata
folder - Download
v1-5-pruned-emaonly.ckpt
from https://huggingface.co/runwayml/stable-diffusion-v1-5/tree/main and save it in thedata
folder
Just download the ckpt
file from any fine-tuned SD (up to v1.5).
- InkPunk Diffusion: https://huggingface.co/Envvi/Inkpunk-Diffusion/tree/main
- Illustration Diffusion (Hollie Mengert): https://huggingface.co/Envvi/Inkpunk-Diffusion/tree/main
Special thanks to the following repositories:
- https://github.com/CompVis/stable-diffusion/
- https://github.com/divamgupta/stable-diffusion-tensorflow
- https://github.com/kjsman/stable-diffusion-pytorch
- https://github.com/huggingface/diffusers/
Install all required modules, then run sd/demo.py. If you have a GPU with > 10 GByte VRAM, set ALLOW_CUDA=True .
Upload sd/demo_colab.ipynb to Google Colab. Enable T4 GPU, set ALLOW_CUDA=True .