add SD weights

app.py

@@ -349,8 +349,8 @@ else:
     # model = instantiate_from_config(config.model)
     
     print('Downloading stable diffusion pretrained weights')
-    SD_PRETRAINED_PATH = 'https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt'
     subprocess.call(['sh', './download.sh'])
+    model = load_model_from_config(config, f"{opt.ckpt}")
 
 model = model.to(device)
 

download.sh

@@ -1,2 +1,3 @@
 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