!nvidia-smi
!apt update && apt dist-upgrade -y
%cd /workspace/
!apt install -y -qq aria2
!aria2c --continue --allow-overwrite --always-resume --summary-interval=10 --disk-cache=64M --min-split-size=8M --max-concurrent-downloads=8 --max-connection-per-server=8 --max-overall-download-limit=0 --max-download-limit=0 --split=8 https://pub-2fdef7a2969f43289c42ac5ae3412fd4.r2.dev/naifu.tar
!aria2c --continue --allow-overwrite --always-resume --summary-interval=10 --disk-cache=64M --min-split-size=8M --max-concurrent-downloads=8 --max-connection-per-server=8 --max-overall-download-limit=0 --max-download-limit=0 --split=8 https://pub-2fdef7a2969f43289c42ac5ae3412fd4.r2.dev/animefull-latest.tar
!echo "Decompressing..."
!tar xf naifu.tar && rm naifu.tar
!echo "Done."
%cd /workspace/naifu
!pip install virtualenv && bash ./setup.sh
!curl -Ls https://github.com/ekzhang/bore/releases/download/v0.4.0/bore-v0.4.0-x86_64-unknown-linux-musl.tar.gz | tar zx -C /usr/bin
!curl -Lo /usr/bin/cloudflared https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-linux-amd64 && chmod +x /usr/bin/cloudflared
%cd /workspace/
!tar xf animefull-latest.tar -C /workspace/naifu/models && rm animefull-latest.tar
!sed -i 's/map_location="cpu"/map_location="cuda"/g' /workspace/naifu/hydra_node/models.py
%cd /workspace/naifu
%env DTYPE=float16
%env CLIP_CONTEXTS=3
%env AMP=1
%env MODEL=stable-diffusion
%env DEV=True
%env MODEL_PATH=models/animefull-latest
%env ENABLE_EMA=1
%env VAE_PATH=models/animevae.pt
%env PENULTIMATE=1
%env PYTHONDONTWRITEBYTECODE=1
%env SAVE_FILES=1
!./venv/bin/python -m uvicorn --host 0.0.0.0 --port=6969 main:app & bore local 6969 --to bore.pub & cloudflared tunnel --url localhost:6969