works at work

.gitignore

@@ -9,3 +9,4 @@ dist
 /env
 a.out
 boxes.jpg
+weights/*

README.md

@@ -162,13 +162,15 @@ PROPROTIP: Set "DEBUG=1" environment variable if you want to see why it's slow.
 
 ### tinygrad supports Stable Diffusion!
 
-Run `LAZY=0 OPT=2 TORCH=1 python3 examples/stable_diffusion.py`
+Run `TORCH=1 python3 examples/stable_diffusion.py`
 
 <p align="center">
   <img src="https://raw.githubusercontent.com/geohot/tinygrad/master/docs/stable_diffusion_by_tinygrad.jpg">
 </p>
 
+<p align="center">
 "a horse sized cat eating a bagel"
+</p>
 
 ### tinygrad supports GANs
 

examples/stable_diffusion.py

@@ -516,7 +516,8 @@ class StableDiffusion:
 
 # this is sd-v1-4.ckpt
 #FILENAME = "/Users/kafka/fun/mps/stable-diffusion/models/ldm/stable-diffusion-v1/model.ckpt"
-FILENAME = "/home/kafka/model.ckpt"
+#FILENAME = "/home/kafka/model.ckpt"
+FILENAME = "weights/sd-v1-4.ckpt"
 
 if __name__ == "__main__":
   Tensor.no_init = True