File size: 9,344 Bytes
87ec67a 4b61e45 87ec67a 2410d08 87ec67a 26badc5 87ec67a 77fad5e 87ec67a 003571d 0341f63 87ec67a 134560e 003571d 134560e 87ec67a ab19329 87ec67a d94078b 003571d d94078b 003571d d94078b 760c504 0cbe40c e3a9852 134560e d94078b ab19329 87ec67a d94078b 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 |
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/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/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/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")
#cards
os.system(f"mkdir /home/user/app/stable-diffusion-webui/extensions/sd-dynamic-prompts/wildcards/")
#negative embeddings
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")
#Lora
os.system(f"mkdir /home/user/app/stable-diffusion-webui/models/Lora/")
os.system(f"git clone https://huggingface.co/galverse/mama-v1 /home/user/app/stable-diffusion-webui/models/Lora/mama")
os.system(f"git clone https://huggingface.co/galverse/mama-v2 /home/user/app/stable-diffusion-webui/models/Lora/mama")
#VAE
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")
#models
os.system(f"wget -c https://huggingface.co/cagliostrolab/animagine-xl-3.1/resolve/main/animagine-xl-3.1.safetensors -P /home/user/app/stable-diffusion-webui/models/Stable-diffusion/")
os.system(f"wget -c https://huggingface.co/cagliostrolab/animagine-xl-3.0/resolve/main/animagine-xl-3.0.safetensors -P /home/user/app/stable-diffusion-webui/models/Stable-diffusion/")
os.system(f"wget -c https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0.safetensors -P /home/user/app/stable-diffusion-webui/models/Stable-diffusion/")
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
|