From 2b1a5a6f7822866861a1d9baed0dafd76c618902 Mon Sep 17 00:00:00 2001
From: Gadersd <gadersd@gmail.com>
Date: Sun, 6 Aug 2023 13:55:20 -0400
Subject: [PATCH] Use cpu for weight dumping

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index efc9fbe..b915476 100644
--- a/README.md
+++ b/README.md
@@ -50,7 +50,7 @@ cd python
 wget https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt
 
 # Extract the weights
-python3 dump.py sd-v1-4.ckpt
+CPU=1 python3 dump.py sd-v1-4.ckpt
 
 # Move the extracted weight folder out
 mv params ..