github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

works at work

Browse Source
This commit is contained in:
George Hotz
2022-09-06 08:06:11 -07:00
parent f683b26eef
commit 682dc64430
3 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/stable_diffusion.py
View File

@@ -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