https://github.com/hkproj/pytorch-stable-diffusion

Stable Diffusion implemented from scratch in PyTorch
https://github.com/hkproj/pytorch-stable-diffusion

diffusion-models latent-diffusion-models paper-implementations pytorch pytorch-implementation stable-diffusion

Last synced: about 6 hours ago
JSON representation

Stable Diffusion implemented from scratch in PyTorch

• Host: GitHub
• URL: https://github.com/hkproj/pytorch-stable-diffusion
• Owner: hkproj
• License: mit
• Created: 2023-09-24T00:00:17.000Z (about 1 year ago)
• Default Branch: main
• Last Pushed: 2024-10-22T08:34:36.000Z (27 days ago)
• Last Synced: 2024-10-23T12:31:26.465Z (25 days ago)
• Topics: diffusion-models, latent-diffusion-models, paper-implementations, pytorch, pytorch-implementation, stable-diffusion
• Language: Jupyter Notebook
• Homepage: https://www.youtube.com/watch?v=ZBKpAp_6TGI
• Size: 1.08 MB
• Stars: 564
• Watchers: 10
• Forks: 127
• Open Issues: 10
• Metadata Files:
  • Readme: README.md
  • License: license.txt

Awesome Lists containing this project

README

        
# 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

## Tested fine-tuned models:

Just download the `ckpt` file from any fine-tuned SD (up to v1.5).

1. InkPunk Diffusion: https://huggingface.co/Envvi/Inkpunk-Diffusion/tree/main

2. Illustration Diffusion (Hollie Mengert): https://huggingface.co/ogkalu/Illustration-Diffusion/tree/main

## Special thanks

Special thanks to the following repositories:

1. https://github.com/CompVis/stable-diffusion/

1. https://github.com/divamgupta/stable-diffusion-tensorflow

1. https://github.com/kjsman/stable-diffusion-pytorch

1. https://github.com/huggingface/diffusers/