Spaces:
Runtime error
Runtime error
File size: 15,219 Bytes
87ec67a 4b61e45 87ec67a 2410d08 87ec67a 26badc5 87ec67a 77fad5e 87ec67a 003571d 0341f63 87ec67a 134560e 003571d 760c504 003571d 134560e 87ec67a ab19329 87ec67a ab19329 55acc76 134560e 003571d 134560e 003571d 134560e 003571d 760c504 003571d 134560e ab19329 87ec67a ab19329 87ec67a 2410d08 e9b766b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 |
import os
from subprocess import getoutput
gpu_info = getoutput('nvidia-smi')
if("A10G" in gpu_info):
os.system(f"pip install -q https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.15/xformers-0.0.15.dev0+4c06c79.d20221205-cp38-cp38-linux_x86_64.whl")
elif("T4" in gpu_info):
os.system(f"apt-get install git-lfs")
os.system(f"pip install -q https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.15/xformers-0.0.15.dev0+1515f77.d20221130-cp38-cp38-linux_x86_64.whl")
os.system(f"git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git /home/user/app/stable-diffusion-webui")
#os.system(f"git clone https://github.com/camenduru/stable-diffusion-webui /home/user/app/stable-diffusion-webui")
os.chdir("/home/user/app/stable-diffusion-webui")
#os.system(f"wget -q https://github.com/camenduru/webui/raw/main/env_patch.py -O /home/user/app/env_patch.py")
#os.system(f"sed -i -e '/import image_from_url_text/r /home/user/app/env_patch.py' /home/user/app/stable-diffusion-webui/modules/ui.py")
#os.system(f"sed -i -e '/(modelmerger_interface, \"Checkpoint Merger\", \"modelmerger\"),/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
#os.system(f"sed -i -e '/(train_interface, \"Train\", \"ti\"),/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
#os.system(f"sed -i -e '/extensions_interface, \"Extensions\", \"extensions\"/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
#os.system(f"sed -i -e '/settings_interface, \"Settings\", \"settings\"/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
#os.system(f'''sed -i -e "s/document.getElementsByTagName('gradio-app')\[0\].shadowRoot/!!document.getElementsByTagName('gradio-app')[0].shadowRoot ? document.getElementsByTagName('gradio-app')[0].shadowRoot : document/g" /home/user/app/stable-diffusion-webui/script.js''')
#os.system(f"sed -i -e 's/ show_progress=False,/ show_progress=True,/g' /home/user/app/stable-diffusion-webui/modules/ui.py")
#os.system(f"sed -i -e 's/shared.demo.launch/shared.demo.queue().launch/g' /home/user/app/stable-diffusion-webui/webui.py")
#os.system(f"sed -i -e 's/ outputs=\[/queue=False, &/g' /home/user/app/stable-diffusion-webui/modules/ui.py")
#os.system(f"sed -i -e 's/ queue=False, / /g' /home/user/app/stable-diffusion-webui/modules/ui.py")
os.system("apt-get install libgoogle-perftools-dev -y")
if "IS_SHARED_UI" in os.environ:
os.system(f"rm -rfv /home/user/app/stable-diffusion-webui/scripts/")
os.system(f"wget -q https://github.com/camenduru/webui/raw/main/shared-config.json -O /home/user/app/shared-config.json")
os.system(f"wget -q https://github.com/camenduru/webui/raw/main/shared-ui-config.json -O /home/user/app/shared-ui-config.json")
os.system(f"wget -q {os.getenv('MODEL_LINK')} -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/{os.getenv('MODEL_NAME')}")
os.system(f"wget -q {os.getenv('VAE_LINK')} -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/{os.getenv('VAE_NAME')}")
os.system(f"wget -q {os.getenv('YAML_LINK')} -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/{os.getenv('YAML_NAME')}")
os.system(f"python launch.py --force-enable-xformers --disable-console-progressbars --enable-console-prompts --ui-config-file /home/user/app/shared-ui-config.json --ui-settings-file /home/user/app/shared-config.json --cors-allow-origins huggingface.co,hf.space --no-progressbar-hiding")
else:
# Please duplicate this space and delete # character in front of the custom script you want to use or add here more custom scripts with same structure os.system(f"wget -q https://CUSTOM_SCRIPT_URL -O /home/user/app/stable-diffusion-webui/scripts/CUSTOM_SCRIPT_NAME.py")
os.system(f"wget -q https://gist.github.com/camenduru/9ec5f8141db9902e375967e93250860f/raw/d0bcf01786f20107c329c03f8968584ee67be12a/run_n_times.py -O /home/user/app/stable-diffusion-webui/scripts/run_n_times.py")
os.system(f"wget https://huggingface.co/Xenos14/Checkpoints/resolve/main/ui-config.json --content-disposition -O /home/user/app/stable-diffusion-webui/ui-config.json")
os.system(f"wget https://huggingface.co/Xenos14/Checkpoints/resolve/main/config.json --content-disposition -O /home/user/app/stable-diffusion-webui/config.json")
# Please duplicate this space and delete # character in front of the extension you want to use or add here more extensions with same structure os.system(f"git clone https://EXTENSION_GIT_URL /home/user/app/stable-diffusion-webui/extensions/EXTENSION_NAME")
os.system(f"git clone https://github.com/Mikubill/sd-webui-controlnet.git /home/user/app/stable-diffusion-webui/extensions/controlnet")
os.system(f"git clone https://github.com/etherealxx/batchlinks-webui /home/user/app/stable-diffusion-webui/extensions/batchlinks-webui")
os.system(f"git clone https://github.com/zanllp/sd-webui-infinite-image-browsing /home/user/app/stable-diffusion-webui/extensions/sd-webui-infinite-image-browsing")
os.system(f"git clone https://github.com/Bing-su/adetailer /home/user/app/stable-diffusion-webui/extensions/adetailer")
os.system(f"git clone https://github.com/hako-mikan/sd-webui-supermerger /home/user/app/stable-diffusion-webui/extensions/sd-webui-supermerger")
os.system(f"git clone https://github.com/aka7774/sd_filer /home/user/app/stable-diffusion-webui/extensions/sd_filer")
os.system(f"git clone https://github.com/arenasys/stable-diffusion-webui-model-toolkit /home/user/app/stable-diffusion-webui/extensions/stable-diffusion-webui-model-toolkit")
os.system(f"git clone https://github.com/civitai/sd_civitai_extension /home/user/app/stable-diffusion-webui/extensions/sd_civitai_extension")
os.system(f"git clone https://github.com/adieyal/sd-dynamic-prompts /home/user/app/stable-diffusion-webui/extensions/sd-dynamic-prompts")
os.system(f"git clone https://github.com/camenduru/stable-diffusion-webui-huggingface /home/user/app/stable-diffusion-webui/extensions/stable-diffusion-webui-huggingface")
os.system(f"git clone https://github.com/BlafKing/sd-civitai-browser-plus /home/user/app/stable-diffusion-webui/extensions/sd-civitai-browser-plus")
os.system(f"git clone https://github.com/DominikDoom/a1111-sd-webui-tagcomplete.git /home/user/app/stable-diffusion-webui/extensions/a1111-sd-webui-tagcomplete")
os.system(f"git clone https://github.com/continue-revolution/sd-webui-animatediff /home/user/app/stable-diffusion-webui/extensions/sd-webui-animatediff")
os.system(f"git clone https://github.com/klimaleksus/stable-diffusion-webui-embedding-merge /home/user/app/stable-diffusion-webui/extensions/stable-diffusion-webui-embedding-merge")
os.system(f"git clone https://github.com/kohya-ss/sd-webui-additional-networks /home/user/app/stable-diffusion-webui/extensions/sd-webui-additional-networks")
#os.system(f"git clone https://github.com/some9000/StylePile /home/user/app/stable-diffusion-webui/extensions/StylePile")
os.system(f"git clone https://github.com/zixaphir/Stable-Diffusion-Webui-Civitai-Helper /home/user/app/stable-diffusion-webui/extensions/Stable-Diffusion-Webui-Civitai-Helper")
os.system(f"git clone https://github.com/mcmonkeyprojects/sd-dynamic-thresholding.git /home/user/app/stable-diffusion-webui/extensions/sd-dynamic-thresholding")
#os.system(f"git clone https /home/user/app/stable-diffusion-webui/extensions/")
#os.system(f"git clone https://github.com/camenduru/stable-diffusion-webui-artists-to-study /home/user/app/stable-diffusion-webui/extensions/stable-diffusion-webui-artists-to-study")
#os.system(f"git clone https://github.com/yfszzx/stable-diffusion-webui-images-browser /home/user/app/stable-diffusion-webui/extensions/stable-diffusion-webui-images-browser")
#os.system(f"git clone https://github.com/deforum-art/deforum-for-automatic1111-webui /home/user/app/stable-diffusion-webui/extensions/deforum-for-automatic1111-webui")
# Please duplicate this space and delete # character in front of the model you want to use or add here more ckpts with same structure os.system(f"wget -q https://CKPT_URL -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/CKPT_NAME.ckpt")
os.system(f"wget -q https://huggingface.co/Xenos14/Checkpoints/resolve/main/XenoMATION-V1.safetensors -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/XenoMATION-V1.safetensors")
#os.system(f"wget https://huggingface.co/Xenos14/TestBed/resolve/main/2024-Real/XenoBABES-V2.safetensors --content-disposition -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/XenoBABES-V2.safetensors")
os.system(f"wget https://huggingface.co/Xenos14/Checkpoints/resolve/main/XenoGASM-MK2.safetensors --content-disposition -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/XenoGASM-MK2.safetensors")
os.system(f"wget https://huggingface.co/Xenos14/TestBed/resolve/main/2024-Real/XenoGRAPH.safetensors --content-disposition -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/XenoGRAPH.safetensors")
os.system(f"wget https://huggingface.co/Xenos14/TestBed/resolve/main/2024-Morph/XenoMORPH-v2.safetensors --content-disposition -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/XenoMORPH-v2.safetensors")
os.system(f"wget https://huggingface.co/Xenos14/XenoREALITY/resolve/main/XenTAI_v1.safetensors --content-disposition -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/XenTAI_v1.safetensors")
os.system(f"wget https://huggingface.co/Xenos14/TestBed/resolve/main/2024-Real/XenoREALITY-V3.safetensors --content-disposition -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/XenoREALITY-V3.safetensors")
os.system(f"wget https://huggingface.co/Xenos14/TestBed/resolve/main/2024-Gen9/XenoENGINE-ArtStyle-v5.safetensors --content-disposition -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/XenoENGINE-ArtStyle-v5.safetensors")
#cards
os.system(f"mkdir /home/user/app/stable-diffusion-webui/extensions/sd-dynamic-prompts/wildcards/")
os.system(f"git clone https://huggingface.co/Xenos14/wildcards /home/user/app/stable-diffusion-webui/extensions/sd-dynamic-prompts/wildcards/Xenos")
os.system(f"rm -r -f /home/user/app/stable-diffusion-webui/extensions/sd-dynamic-prompts/wildcards/Xenos/.git")
os.system(f"git clone https://github.com/vorstcavry/embeddings /home/user/app/stable-diffusion-webui/embeddings/negative")
os.system(f"rm -r -f /home/user/app/stable-diffusion-webui/embeddings/negative/.git")
os.system(f"git clone https://huggingface.co/Xenos14/embeddings /home/user/app/stable-diffusion-webui/embeddings/!Xenos")
os.system(f"rm -r -f /home/user/app/stable-diffusion-webui/embeddings/!Xenos/.git")
os.system(f"mkdir /home/user/app/stable-diffusion-webui/models/Lora/")
os.system(f"git clone https://huggingface.co/Xenos14/Loras /home/user/app/stable-diffusion-webui/models/Lora/Xenos")
os.system(f"rm -r -f /home/user/app/stable-diffusion-webui/models/Lora/Xenos/.git")
os.system(f"wget -q https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.ckpt -O /home/user/app/stable-diffusion-webui/models/VAE/vae-ft-mse-840000-ema-pruned.ckpt")
#os.system(f"wget -q https://huggingface.co/nitrosocke/Arcane-Diffusion/resolve/main/arcane-diffusion-v3.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/arcane-diffusion-v3.ckpt")
#os.system(f"wget -q https://huggingface.co/DGSpitzer/Cyberpunk-Anime-Diffusion/resolve/main/Cyberpunk-Anime-Diffusion.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/Cyberpunk-Anime-Diffusion.ckpt")
#os.system(f"wget -q https://huggingface.co/prompthero/midjourney-v4-diffusion/resolve/main/mdjrny-v4.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/mdjrny-v4.ckpt")
#os.system(f"wget -q https://huggingface.co/nitrosocke/mo-di-diffusion/resolve/main/moDi-v1-pruned.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/moDi-v1-pruned.ckpt")
#os.system(f"wget -q https://huggingface.co/Fictiverse/Stable_Diffusion_PaperCut_Model/resolve/main/PaperCut_v1.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/PaperCut_v1.ckpt")
#os.system(f"wget -q https://huggingface.co/lilpotat/sa/resolve/main/samdoesarts_style.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/samdoesarts_style.ckpt")
#os.system(f"wget -q https://huggingface.co/hakurei/waifu-diffusion-v1-3/resolve/main/wd-v1-3-float32.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/wd-v1-3-float32.ckpt")
#os.system(f"wget -q https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/sd-v1-4.ckpt")
#os.system(f"wget -q https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/v1-5-pruned-emaonly.ckpt")
#os.system(f"wget -q https://huggingface.co/runwayml/stable-diffusion-inpainting/resolve/main/sd-v1-5-inpainting.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/sd-v1-5-inpainting.ckpt")
#os.system(f"wget -q https://huggingface.co/Linaqruf/anything-v3.0/resolve/main/Anything-V3.0-pruned.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/Anything-V3.0-pruned.ckpt")
#os.system(f"wget -q https://huggingface.co/Linaqruf/anything-v3.0/resolve/main/Anything-V3.0.vae.pt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/Anything-V3.0-pruned.vae.pt")
#os.system(f"wget -q https://huggingface.co/stabilityai/stable-diffusion-2/resolve/main/768-v-ema.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/768-v-ema.ckpt")
#os.system(f"wget -q https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/768-v-ema.yaml")
#os.system(f"wget -q https://r2.kamiya-b.me/dreambooth_lib/akakura-sn.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/akakura-sn.ckpt")
#os.system(f"wget -q https://huggingface.co/stabilityai/stable-diffusion-2-1/resolve/main/v2-1_768-ema-pruned.ckpt -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/v2-1_768-ema-pruned.ckpt")
os.system(f"wget -q https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/v2-1_768-ema-pruned.yaml")
os.system(f"python launch.py --enable-console-prompts --cors-allow-origins huggingface.co,hf.space --no-progressbar-hiding --api --skip-torch-cuda-test --disable-safe-unpickle --skip-torch-cuda-test --no-half --listen --enable-insecure-extension-access")
#--ui-config-file /home/user/app/ui-config.json --ui-settings-file /home/user/app/config.json
|