Update app.py

app.py

@@ -1,21 +1,43 @@
-import os
-os.kill(os.getpid(), 9) # This will crash Colab (required, everything will still be intact so dont worry)
-
-git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui /content/stable-diffusion-webui
+!curl -Lo memfix.zip https://github.com/nolanaatama/sd-webui/raw/main/memfix.zip
+!unzip /content/memfix.zip
+!apt install -qq libunwind8-dev
+!dpkg -i *.deb
+%env LD_PRELOAD=libtcmalloc.so
+!rm *
+!pip install --upgrade fastapi==0.90.1
+!pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 torchtext==0.14.1 torchdata==0.5.1 --extra-index-url https://download.pytorch.org/whl/cu116 -U
+!git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui
+!git clone https://github.com/nolanaatama/sd-webui-tunnels /content/stable-diffusion-webui/extensions/sd-webui-tunnels
+!git clone https://github.com/Mikubill/sd-webui-controlnet /content/stable-diffusion-webui/extensions/sd-webui-controlnet
+!git clone https://github.com/hnmr293/posex /content/stable-diffusion-webui/extensions/posex
+!git clone https://github.com/yfszzx/stable-diffusion-webui-images-browser /content/stable-diffusion-webui/extensions/stable-diffusion-webui-images-browser
+# Model Code
+!curl -Lo /content/stable-diffusion-webui/models/Stable-diffusion/anythingv5.safetensors https://huggingface.co/nolanaatama/nythngv5/resolve/main/nythngv5.safetensors
+# ControlNet
+!curl -Lo /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models/control_canny.safetensors https://huggingface.co/webui/ControlNet-modules-safetensors/resolve/main/control_canny-fp16.safetensors
+!curl -Lo /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models/control_depth.safetensors https://huggingface.co/webui/ControlNet-modules-safetensors/resolve/main/control_depth-fp16.safetensors
+!curl -Lo /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models/control_hed-fp16.safetensors https://huggingface.co/webui/ControlNet-modules-safetensors/resolve/main/control_hed-fp16.safetensors
+!curl -Lo /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models/control_mlsd-fp16.safetensors https://huggingface.co/webui/ControlNet-modules-safetensors/resolve/main/control_mlsd-fp16.safetensors
+!curl -Lo /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models/control_normal-fp16.safetensors https://huggingface.co/webui/ControlNet-modules-safetensors/resolve/main/control_normal-fp16.safetensors
+!curl -Lo /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models/control_openpose-fp16.safetensors https://huggingface.co/webui/ControlNet-modules-safetensors/resolve/main/control_openpose-fp16.safetensors
+!curl -Lo /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models/control_scribble-fp16.safetensors https://huggingface.co/webui/ControlNet-modules-safetensors/resolve/main/control_scribble-fp16.safetensors
+!curl -Lo /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models/control_seg-fp16.safetensors https://huggingface.co/webui/ControlNet-modules-safetensors/resolve/main/control_seg-fp16.safetensors
+!curl -Lo /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models/t2iadapter_canny_sd14v1.pth https://huggingface.co/TencentARC/T2I-Adapter/resolve/main/models/t2iadapter_canny_sd14v1.pth
+!curl -Lo /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models/t2iadapter_color_sd14v1.pth https://huggingface.co/TencentARC/T2I-Adapter/resolve/main/models/t2iadapter_color_sd14v1.pth
+!curl -Lo /content/stable-diffusion-wcebui/extensions/sd-webui-controlnet/models/t2iadapter_depth_sd14v1.pth https://huggingface.co/TencentARC/T2I-Adapter/resolve/main/models/t2iadapter_depth_sd14v1.pth
+!curl -Lo /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models/t2iadapter_keypose_sd14v1.pth https://huggingface.co/TencentARC/T2I-Adapter/resolve/main/models/t2iadapter_keypose_sd14v1.pth
+!curl -Lo /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models/t2iadapter_openpose_sd14v1.pth https://huggingface.co/TencentARC/T2I-Adapter/resolve/main/models/t2iadapter_openpose_sd14v1.pth
+!curl -Lo /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models/t2iadapter_seg_sd14v1.pth https://huggingface.co/TencentARC/T2I-Adapter/resolve/main/models/t2iadapter_seg_sd14v1.pth
+!curl -Lo /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models/t2iadapter_sketch_sd14v1.pth https://huggingface.co/TencentARC/T2I-Adapter/resolve/main/models/t2iadapter_sketch_sd14v1.pth
+!curl -Lo /content/stable-diffusion-webui/extensions/sd-webui-controlnet/models/t2iadapter_style_sd14v1.pth https://huggingface.co/TencentARC/T2I-Adapter/resolve/main/models/t2iadapter_style_sd14v1.pth
+import shutil
+shutil.rmtree('/content/stable-diffusion-webui/embeddings')
 %cd /content/stable-diffusion-webui
-
-#@title Normal 1.4 model
-# get a token from https://huggingface.co/settings/tokens
-user_token = "" #@param {type:"string"}
-user_header = f"\"Authorization: Bearer {user_token}\""
-!wget --header={user_header} https://huggingface.co/nolanaatama/nythngv5/resolve/main/nythngv5.safetensors -O /content/stable-diffusion-webui/models/Stable-diffusion/anythingv5.safetensors
-
-!COMMANDLINE_ARGS="--exit" REQS_FILE="requirements.txt" python launch.py
-
-import os
-os.kill(os.getpid(), 9) # This will crash Colab (required, everything will still be intact so dont worry)
-
-
-
-
-
+!git checkout 0cc0ee1
+!git clone https://huggingface.co/nolanaatama/embeddings
+%cd /content/stable-diffusion-webui/models
+!git clone https://huggingface.co/nolanaatama/ESRGAN
+%cd /content/stable-diffusion-webui
+# Web UI tunnel
+!COMMANDLINE_ARGS="--share --disable-safe-unpickle --no-half-vae --xformers --reinstall-xformers --enable-insecure-extension --gradio-queue --remotemoe" REQS_FILE="requirements.txt" python launch.py
+# If remotemoe failed to start, change '--remotemoe' to '--cloudflared' on the COMMANDLINE_ARGS line above to use cloudflare tunnel