Use cpu for weight dumping

This commit is contained in:
Gadersd
2023-08-06 13:55:20 -04:00
committed by Ben_Kosytorz
parent c596eab0e3
commit 2b1a5a6f78

View File

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