Skip to content

Commit f48b72e

Browse files
authored
Update README.md add demo
1 parent 9bbebe4 commit f48b72e

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

README.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,9 +11,17 @@ This tiny self-contained code base allows you to
1111
* Rebuild the Stable Diffusion Model in a single Python script.
1212
* Train your toy version of stable diffusion on classic datasets like MNIST, CelebA
1313
![](https://scholar.harvard.edu/sites/scholar.harvard.edu/files/styles/os_files_xxlarge/public/binxuw/files/stablediffusion_overview.jpg?m=1667438590&itok=n2gM0Xba)
14+
1415
## Colab notebooks
1516
* Playing with Stable Diffusion and inspecting the internal architecture of the models. [Open in Colab](https://colab.research.google.com/drive/1TvOlX2_l4pCBOKjDI672JcMm4q68sKrA?usp=sharing)
1617
* Build your own Stable Diffusion UNet model from scratch in a notebook. (with < 300 lines of codes!) [Open in Colab](https://colab.research.google.com/drive/1mm67_irYu3qU3hnfzqK5yQC38Fd5UFam?usp=sharing)
1718
* [Self contained script](https://github.com/Animadversio/DiffusionFromScratch/blob/master/StableDiff_UNet_model.py)
1819
* [Unit tests](https://github.com/Animadversio/DiffusionFromScratch/blob/master/StableDiff_UNet_unittest.py)
1920
* Build a Diffusion model (with UNet + cross attention) and train it to generate MNIST images based on the "text prompt". [Open in Colab (exercise)](https://colab.research.google.com/drive/1Y5wr91g5jmpCDiX-RLfWL1eSBWoSuLqO?usp=sharing) [Open in Colab (answer)](https://colab.research.google.com/drive/1_MEFfBdOI06GAuANrs1b8L-BBLn3x-ZJ?usp=sharing)
21+
22+
23+
## Demo Outputs
24+
[![IMAGE ALT TEXT HERE](https://img.youtube.com/vi/SmY7vMNen2w/0.jpg)](https://www.youtube.com/watch?v=SmY7vMNen2w)
25+
26+
[Music video](https://github.com/nateraw/stable-diffusion-videos) generated from Stable Diffusion.
27+

0 commit comments

Comments
 (0)