mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-13 08:28:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add step for setting up Stable Diffusion (#452)

Browse Source
This commit is contained in:
Nicolai Stoianov
2023-01-07 17:40:12 +01:00
committed by GitHub
parent 4fb97b8de0
commit 8dbf76268d
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -172,6 +172,8 @@ PROPROTIP: Set "DEBUG=1" environment variable if you want to see why it's slow.
### tinygrad supports Stable Diffusion!
You might need to download the [weight](https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt) of Stable Diffusion and put it into weights/
Run `TORCH=1 python3 examples/stable_diffusion.py`
(or without torch: `OPT=2 OPENCL=1 python3 examples/stable_diffusion.py`)