From 682dc644303440eaab492a16cf0326aa0d7eb877 Mon Sep 17 00:00:00 2001
From: George Hotz <george@comma.ai>
Date: Tue, 6 Sep 2022 08:06:11 -0700
Subject: [PATCH] works at work

---
 .gitignore                   | 1 +
 README.md                    | 4 +++-
 examples/stable_diffusion.py | 3 ++-
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/.gitignore b/.gitignore
index e72ad10e31..4911dd7c18 100644
--- a/.gitignore
+++ b/.gitignore
@@ -9,3 +9,4 @@ dist
 /env
 a.out
 boxes.jpg
+weights/*
diff --git a/README.md b/README.md
index 8615b9d50e..1ba70bf7e5 100644
--- a/README.md
+++ b/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
 
diff --git a/examples/stable_diffusion.py b/examples/stable_diffusion.py
index 0aea05d16d..7bfe4a1622 100644
--- a/examples/stable_diffusion.py
+++ b/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