Spaces:
Running
Running
| import os | |
| user_home = r"/home/user/app" | |
| os.chdir(user_home) | |
| #clone stable-diffusion-webui repo | |
| print("cloning stable-diffusion-webui repo") | |
| os.system("git clone \"https://github.com/AUTOMATIC1111/stable-diffusion-webui.git\" "+user_home+r"/stable-diffusion-webui") | |
| #install extensions | |
| print("installing extensions") | |
| os.system(r"git clone !git clone https://huggingface.co./embed/negative "+user_home+"/stable-diffusion-webui/embeddings/negative") | |
| os.system(r"git clone https://huggingface.co./embed/lora "+user_home+"/stable-diffusion-webui/models/Lora/positive") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./embed/upscale/resolve/main/4x-UltraSharp.pth -d "+user_home+"/stable-diffusion-webui/models/ESRGAN -o 4x-UltraSharp.ptharia2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./embed/upscale/resolve/main/4x-UltraSharp.pth -d "+user_home+"/stable-diffusion-webui/models/ESRGAN -o 4x-UltraSharp.pth") | |
| os.system(r"wget https://raw.githubusercontent.com/camenduru/stable-diffusion-webui-scripts/main/run_n_times.py -O "+user_home+"/stable-diffusion-webui/scripts/run_n_times.py") | |
| os.system(r"git clone https://github.com/AlUlkesh/stable-diffusion-webui-images-browser "+user_home+"/stable-diffusion-webui/extensions/stable-diffusion-webui-images-browser") | |
| os.system(r"git clone https://github.com/camenduru/stable-diffusion-webui-huggingface "+user_home+"/stable-diffusion-webui/extensions/stable-diffusion-webui-huggingface") | |
| os.system(r"git clone https://github.com/Mikubill/sd-webui-controlnet "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet") | |
| os.system(r"git clone https://github.com/jexom/sd-webui-depth-lib "+user_home+"/stable-diffusion-webui/extensions/sd-webui-depth-lib") | |
| os.system(r"git clone https://github.com/hnmr293/posex "+user_home+"/stable-diffusion-webui/extensions/posex") | |
| os.system(r"git clone https://github.com/nonnonstop/sd-webui-3d-open-pose-editor "+user_home+"/stable-diffusion-webui/extensions/sd-webui-3d-open-pose-editor") | |
| os.system(r"git clone https://github.com/imrayya/stable-diffusion-webui-Prompt_Generator "+user_home+"/stable-diffusion-webui/extensions/sd-webui-Prompt_Generator") | |
| os.chdir(os.path.join(user_home,r"stable-diffusion-webui")) | |
| os.system(r"git reset --hard") | |
| os.system(r"git -C "+user_home+"/stable-diffusion-webui/repositories/stable-diffusion-stability-ai reset --hard") | |
| #download ControlNet models | |
| print("downloading ControlNet models") | |
| os.system("wget https://github.com/aria2/aria2/releases/download/release-1.36.0/aria2-1.36.0-aarch64-linux-android-build1.zip") | |
| os.system("unzip aria2-1.36.0-aarch64-linux-android-build1.zip") | |
| os.system("mv aria2c /usr/local/bin") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/control_v11e_sd15_ip2p_fp16.safetensors -d "+user_home+""+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11e_sd15_ip2p_fp16.safetensors") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/control_v11e_sd15_shuffle_fp16.safetensors -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11e_sd15_shuffle_fp16.safetensors") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_canny_fp16.safetensors -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_canny_fp16.safetensors") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/control_v11f1p_sd15_depth_fp16.safetensors -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11f1p_sd15_depth_fp16.safetensors") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_inpaint_fp16.safetensors -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_inpaint_fp16.safetensors") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_lineart_fp16.safetensors -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_lineart_fp16.safetensors") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_mlsd_fp16.safetensors -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_mlsd_fp16.safetensors") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_normalbae_fp16.safetensors -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_normalbae_fp16.safetensors") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_openpose_fp16.safetensors -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_openpose_fp16.safetensors") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_scribble_fp16.safetensors -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_scribble_fp16.safetensors") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_seg_fp16.safetensors -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_seg_fp16.safetensors") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_softedge_fp16.safetensors -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_softedge_fp16.safetensors") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15s2_lineart_anime_fp16.safetensors -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15s2_lineart_anime_fp16.safetensors") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/control_v11f1e_sd15_tile_fp16.safetensors -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11f1e_sd15_tile_fp16.safetensors") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/raw/main/control_v11e_sd15_ip2p_fp16.yaml -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11e_sd15_ip2p_fp16.yaml") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/raw/main/control_v11e_sd15_shuffle_fp16.yaml -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11e_sd15_shuffle_fp16.yaml") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_canny_fp16.yaml -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_canny_fp16.yaml") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/raw/main/control_v11f1p_sd15_depth_fp16.yaml -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11f1p_sd15_depth_fp16.yaml") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_inpaint_fp16.yaml -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_inpaint_fp16.yaml") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_lineart_fp16.yaml -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_lineart_fp16.yaml") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_mlsd_fp16.yaml -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_mlsd_fp16.yaml") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_normalbae_fp16.yaml -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_normalbae_fp16.yaml") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_openpose_fp16.yaml -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_openpose_fp16.yaml") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_scribble_fp16.yaml -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_scribble_fp16.yaml") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_seg_fp16.yaml -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_seg_fp16.yaml") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_softedge_fp16.yaml -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15_softedge_fp16.yaml") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15s2_lineart_anime_fp16.yaml -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11p_sd15s2_lineart_anime_fp16.yaml") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/raw/main/control_v11f1e_sd15_tile_fp16.yaml -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o control_v11f1e_sd15_tile_fp16.yaml") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/t2iadapter_style_sd14v1.pth -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_style_sd14v1.pth") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/t2iadapter_sketch_sd14v1.pth -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_sketch_sd14v1.pth") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/t2iadapter_seg_sd14v1.pth -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_seg_sd14v1.pth") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/t2iadapter_openpose_sd14v1.pth -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_openpose_sd14v1.pth") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/t2iadapter_keypose_sd14v1.pth -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_keypose_sd14v1.pth") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/t2iadapter_depth_sd14v1.pth -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_depth_sd14v1.pth") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/t2iadapter_depth_sd14v1.pth -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_depth_sd14v1.pth") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/t2iadapter_canny_sd14v1.pth -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_canny_sd14v1.pth") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/t2iadapter_canny_sd15v2.pth -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_canny_sd15v2.pth") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/t2iadapter_depth_sd15v2.pth -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_depth_sd15v2.pth") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/t2iadapter_sketch_sd15v2.pth -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_sketch_sd15v2.pth") | |
| os.system(r"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co./ckpt/ControlNet-v1-1/resolve/main/t2iadapter_zoedepth_sd15v1.pth -d "+user_home+"/stable-diffusion-webui/extensions/sd-webui-controlnet/models -o t2iadapter_zoedepth_sd15v1.pth") | |
| try: | |
| os.mkdir(os.path.join(user_home,"stable-diffusion-webui/models/Stable-diffusion")) | |
| except(FileExistsError): | |
| print("exist") | |
| print("downloading model") | |
| os.system(f"wget -q https://huggingface.co./FredZhang7/paint-journey-v2/resolve/main/paint_journey_v2_cpu_only.ckpt -O {user_home}/stable-diffusion-webui/models/Stable-diffusion/paint_journey_v2_cpu_only.ckpt") | |
| os.system(f"wget -q https://huggingface.co./FredZhang7/paint-journey-v2/resolve/main/paint_journey_v2.vae.pt -O {user_home}/stable-diffusion-webui/models/Stable-diffusion/paint_journey_v2_cpu_only.vae.pt") | |
| #strt webui | |
| print("Done\nStarting Webui...") | |
| os.system(r"python3 launch.py --precision full --ui-config-file /home/user/app/ui-config.json --ui-settings-file /home/user/app/config.json --no-half --no-half-vae --enable-insecure-extension-access --medvram --skip-torch-cuda-test ") | |
| del os ,user_home |