add SD weights

app.py

@@ -350,7 +350,7 @@ else:
     
     print('Downloading stable diffusion pretrained weights')
     subprocess.call(['sh', './download.sh'])
-    model = load_model_from_config(config, f"{opt.ckpt}")
+    model = load_model_from_config(config, "model.ckpt")
 
 model = model.to(device)
 

download.sh

@@ -1,3 +1,2 @@
 wget https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt
-mkdir ./stable-diffusion/models/ldm/stable-diffusion-v1/
-mv v1-5-pruned-emaonly.ckpt ./stable-diffusion/models/ldm/stable-diffusion-v1/model.ckpt
+mv v1-5-pruned-emaonly.ckpt model.ckpt