Implementation of Stable Diffusion in PyTorch, for personal interest and learning purpose.
Weights are stored on a huggingface hub repository and automatically downloaded and cached at runtime.
The easiest way to give it a shot is using one of the following Colab Notebooks:
| textual prompt | output |
|---|---|
| "Italy with cyberpunk style" |  |
| "Galileo in Disney style" |  |
| "DSLR photograph of an astronut riding a horse" |  |
| "Realistic Gandalf playing the piano" |  |
| "Eren from Attack on Titan drinking wine" |  |
The implementation is based on this repository, which is based on these repositories:
- divamgupta/stable-diffusion-tensorflow
- CompVis/stable-diffusion
- huggingface/transformers
- huggingface/diffusers
- AUTOMATIC1111/stable-diffusion-webui
- the
ClipTokenizerhas been rewritten to exploit LRU Caching - the weights and vocabulary download process has entirely automated so that no manual action is required from the user
- add a progress bar to track the generation process and provided a generation script
- the code has been