Use cpu for weight dumping
This commit is contained in:
@@ -50,7 +50,7 @@ cd python
|
|||||||
wget https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt
|
wget https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt
|
||||||
|
|
||||||
# Extract the weights
|
# 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
|
# Move the extracted weight folder out
|
||||||
mv params ..
|
mv params ..
|
||||||
|
|||||||
Reference in New Issue
Block a user