2#!/bin/bash

fuser -k 3000/tcp

# Update package lists
apt update

cd /workspace

#mv /workspace/stable-diffusion-webui/models/Stable-diffusion/sd_xl_base_1.0.safetensors /workspace/

#mv /workspace/stable-diffusion-webui/models/Stable-diffusion/sd_xl_refiner_1.0.safetensors /workspace/

rm -r stable-diffusion-webui

git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui

cd stable-diffusion-webui

wget https://huggingface.co/MonsterMMORPG/SECourses/resolve/main/relauncher.py

git checkout 394ffa7b0a7fff3ec484bcd084e673a8b301ccc8

#mv /workspace/sd_xl_base_1.0.safetensors /workspace/stable-diffusion-webui/models/Stable-diffusion/

#mv /workspace/sd_xl_refiner_1.0.safetensors /workspace/stable-diffusion-webui/models/Stable-diffusion/

rm -r webui-user.sh

wget -O /workspace/stable-diffusion-webui/webui-user.sh https://huggingface.co/MonsterMMORPG/SECourses/raw/main/webui-user-v2.sh

git clone https://github.com/d8ahazard/sd_dreambooth_extension /workspace/stable-diffusion-webui/extensions/sd_dreambooth_extension

cd /workspace/stable-diffusion-webui/extensions/sd_dreambooth_extension

git checkout dc413a14379b165355502d9f65856c40a4bb5b6f

#wget https://huggingface.co/stabilityai/sdxl-vae/resolve/main/sdxl_vae.safetensors -O /workspace/stable-diffusion-webui/models/VAE/sd_xl_base_1.0.safetensors

wget -O /workspace/stable-diffusion-webui/models/VAE/best_SD_1_5_VAE.ckpt https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.ckpt

wget https://huggingface.co/SG161222/Realistic_Vision_V5.1_noVAE/resolve/main/Realistic_Vision_V5.1.safetensors -O /workspace/stable-diffusion-webui/models/Stable-diffusion/Realistic_Vision_V5.1.safetensors

wget https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned.safetensors -O /workspace/stable-diffusion-webui/models/Stable-diffusion/v1-5-pruned.safetensors

cd /workspace/stable-diffusion-webui

wget https://gist.githubusercontent.com/FurkanGozukara/632650d77ed07e88bc5603597a746520/raw/855e2f06e27442f51d0bf0ac25843651697bb26c/requirements_versions.txt -O "requirements_versions.txt"

python relauncher.py