PyTorch Stable Diffusion

PyTorch implementation of Stable Diffusion from scratch

Download weights and tokenizer files:

1. Download vocab.json and merges.txt from https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5/tree/main/tokenizer and save them in the data folder
2. Download v1-5-pruned-emaonly.ckpt from https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5/tree/main and save it in the data folder

Further Reading:

1. Paper: https://arxiv.org/pdf/1503.03585.pdf
2. Paper: https://arxiv.org/pdf/2006.11239.pdf
3. Paper: https://arxiv.org/pdf/2102.09672.pdf
4. Paper: https://arxiv.org/pdf/2105.05233.pdf