!curl -Lo fixmem.zip https://github.com/devchayon/aaii/raw/main/fixmem.zip
!unzip /content/fixmem.zip
!apt -y update -qq
%env LD_PRELOAD=/content/libtcmalloc_minimal.so.4

!curl -Lo aaii.zip https://huggingface.co/devck/aaii/resolve/main/aaifix.zip
!unzip /content/aaii.zip

# Extensions Section
!git clone https://github.com/devchayon/aaii-tunnels /content/aaii/extensions/aaii-tunnels
!git clone https://github.com/devchayon/aaii-controlnet /content/aaii/extensions/aaii-controlnet
!git clone https://github.com/devchayon/aaii-tagcomplete /content/aaii/extensions/aaii-tagcomplete
!git clone https://github.com/devchayon/aaii-locon /content/aaii/extensions/aaii-locon

# Additional Extensions
!mkdir /content/aaii/models/ESRGAN
!curl -Lo /content/aaii/extensions/aaii-images-browser.zip https://huggingface.co/devck/aaii/resolve/main/aaii-images-browser.zip
!curl -Lo /content/aaii/embeddings/embeddings.zip https://huggingface.co/devck/aaii/resolve/main/embeddings.zip
!curl -Lo /content/aaii/models/ESRGAN/upscalers.zip https://huggingface.co/devck/aaii/resolve/main/upscalers.zip
%cd /content/aaii/extensions
!unzip /content/aaii/extensions/aaii-images-browser.zip
%cd /content/aaii/embeddings
!unzip /content/aaii/embeddings/embeddings.zip
%cd /content/aaii/models/ESRGAN
!unzip /content/aaii/models/ESRGAN/upscalers.zip
!rm upscalers.zip
%cd /content

# Model Code
#Anypest
!curl -Lo /content/aaii/models/Stable-diffusion/anypastel.safetensors https://huggingface.co/nolanaatama/nypstl/resolve/main/nypstl.safetensors
!curl -Lo /content/aaii/models/VAE/anypastel.vae.pt https://huggingface.co/nolanaatama/nypstl/resolve/main/nypstl.vae.pt

#Aoe3
!curl -Lo /content/aaii/models/Stable-diffusion/abyssorangemix3.safetensors https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/Models/AbyssOrangeMix3/AOM3_orangemixs.safetensors
!curl -Lo /content/aaii/models/VAE/abyssorangemix3.vae.pt https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/VAEs/orangemix.vae.pt


# ControlNet
# Remove '#' from the beginning of the line(s) below to download the selected ControlNet model(s)
!curl -Lo /content/aaii/extensions/aaii-controlnet/models/control_v11f1p_sd15_depth.safetensors https://huggingface.co/devck/aaii/resolve/main/control_v11p_sd15_depth_fp16.safetensors
!curl -Lo /content/aaii/extensions/aaii-controlnet/models/control_v11p_sd15s2_lineart_anime.safetensors https://huggingface.co/devck/aaii/resolve/main/control_v11p_sd15s2_lineart_anime_fp16.safetensors

!rm aaii.zip
%cd /content/aaii

!pip install fastapi==0.90

#
!COMMANDLINE_ARGS="--share --disable-safe-unpickle --no-half-vae --xformers --enable-insecure-extension --gradio-queue" REQS_FILE="requirements.txt" python launch.py
# Use this command below to use cloudflare tunnel
#!COMMANDLINE_ARGS="--disable-safe-unpickle --no-half-vae --xformers --enable-insecure-extension --gradio-queue --cloudflared" REQS_FILE="requirements.txt" python launch.py