Update scripts/mainrunpodA1111.py

scripts/mainrunpodA1111.py

@@ -148,6 +148,12 @@ def mdl(Original_Model_Version, Path_to_MODEL, MODEL_LINK):
     
     src=getsrc(MODEL_LINK)  
 
+    if os.path.exists('/workspace/sd/stable-diffusion-webui/models/VAE'):
+        call('ln -s /workspace/auto-VAE/* /workspace/sd/stable-diffusion-webui/models/VAE', shell=True)
+    else:
+        call('mkdir -p /workspace/sd/stable-diffusion-webui/models/VAE', shell=True)
+        call('ln -s /workspace/auto-VAE/* /workspace/sd/stable-diffusion-webui/models/VAE', shell=True)
+
     if not os.path.exists('/workspace/sd/stable-diffusion-webui/models/Stable-diffusion/SDv1-5.ckpt'):
         call('ln -s /workspace/auto-models/* /workspace/sd/stable-diffusion-webui/models/Stable-diffusion', shell=True)
     
@@ -413,6 +419,9 @@ def sd(User, Password, model):
     call("sed -i 's@\[\"sd_model_checkpoint\"\]@\[\"sd_model_checkpoint\", \"sd_vae\", \"CLIP_stop_at_last_layers\", \"inpainting_mask_weight\", \"initial_noise_multiplier\"\]@g' /workspace/sd/stable-diffusion-webui/modules/shared.py", shell=True)
     
     call("sed -i 's@print(\"No module.*@@' /workspace/sd/stablediffusion/ldm/modules/diffusionmodules/model.py", shell=True)
+
+    call('wget -O /workspace/sd/stable-diffusion-webui/config.json https://huggingface.co/spaces/stlaurentjr/RNPD/raw/main/config/config.json', shell=True)
+    
     os.chdir('/workspace/sd/stable-diffusion-webui')
     clear_output()