Test / app.py
Blackwolf2's picture
Update app.py
736a5e1
import os
os.chdir(f"/home/user/app")
os.system(f"apt -y update -qq")
os.system(f"wget https://github.com/camenduru/gperftools/releases/download/v1.0/libtcmalloc_minimal.so.4 -O /home/user/app/libtcmalloc_minimal.so.4")
os.system(f"env LD_PRELOAD=/home/user/app/libtcmalloc_minimal.so.4")
os.system(f"apt -y install -qq aria2")
os.system(f"pip install -q xformers==0.0.20 triton==2.0.0 -U")
os.system(f"git clone -b v2.6 https://dagshub.com/camenduru/ui")
os.system(f"git clone https://github.com/camenduru/control /home/user/app/ui/extensions/control")
os.system(f"cd /home/user/app/ui")
os.system(f"git reset --hard")
os.system(f"git -C /home/user/app/ui/repositories/stable-diffusion-stability-ai reset --hard")
os.system(f"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/sd_xl_base_1.0/resolve/main/sd_xl_base_1.0_0.9vae.safetensors -d /home/user/app/ui/models/Stable-diffusion -o sd_xl_base_1.0.safetensors")
os.system(f"python launch.py --cors-allow-origins=* --listen --xformers --enable-insecure-extension-access --theme dark --gradio-queue --multiple")