File size: 10,535 Bytes
17a16a3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4c0989a
17a16a3
 
 
 
 
4c0989a
 
 
 
 
 
 
17a16a3
 
 
 
4c0989a
17a16a3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4c0989a
17a16a3
 
4c0989a
17a16a3
 
 
 
4c0989a
17a16a3
4c0989a
 
 
 
 
17a16a3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
#!/bin/bash
set -euo pipefail

function download-model() {
    local _option=$1
    local _filename=$2
    local _url=$3
    local _dir

    ! [ $# -eq 3 ] && (echo "usage: "; for o in checkpoint lora vae control-net embedding; do echo "  \$ download-model --$o <filename> <url>"; done) || true
    [ $# -eq 0 ] && return 0 || ! [ $# -eq 3 ] && (echo ""; echo "error - invalid number of arguments (expected 3, received $#)"; echo -n "\$ download-model $1"; (for arg in "${@: 2}"; do echo -n " \"${arg//\"/\\\"}\""; done) && echo "") && return 1 || true

    case ${_option,,} in
        --checkpoint) _dir="/app/stable-diffusion-webui/models/Stable-diffusion";;
        --lora) _dir="/app/stable-diffusion-webui/extensions/sd-webui-additional-networks/models/LoRA";;
        --vae) _dir="/app/stable-diffusion-webui/models/VAE";;
        --control-net) _dir="/app/stable-diffusion-webui/models/ControlNet";;
        --embedding) _dir="/app/stable-diffusion-webui/embeddings";;

        *) echo "error - unknown first argument: '$1' (valid options are --checkpoint, --lora, --vae, --control-net or --embedding):"; echo "\$ download-model $1 \"$2\" \"$3\""; return 1;;
    esac

    echo "\$ download-model $_option \"$2\" \"$3\"" ; echo ""
    aria2c --console-log-level=error -c -x 16 -s 16 -k 1M $_url -d $_dir -o $_filename && echo ""
}

## ----------------------------

## Adds a header to the webui on Hugging Face Spaces.
sed -i -e '/demo:/r /app/stable-diffusion-webui/header_patch.py' /app/stable-diffusion-webui/modules/ui.py

## ----------------------------

## Installing less models if $IS_SHARED_UI environment variable is set.
if [ ${IS_SHARED_UI:-0} != 0 ]; then
    download-model --checkpoint "v1-5-pruned-emaonly.safetensors" "https://huggingface.co./runwayml/stable-diffusion-v1-5/resolve/39593d5650112b4cc580433f6b0435385882d819/v1-5-pruned-emaonly.safetensors"
    download-model --checkpoint "v1-5-pruned-emaonly.yaml" "https://huggingface.co./runwayml/stable-diffusion-v1-5/resolve/39593d5650112b4cc580433f6b0435385882d819/v1-inference.yaml"
    download-model --control-net "cldm_v15.yaml" "https://huggingface.co./webui/ControlNet-modules-safetensors/resolve/87c3affbcad3baec52ffe39cac3a15a94902aed3/cldm_v15.yaml"
    download-model --control-net "control_canny-fp16.safetensors" "https://huggingface.co./webui/ControlNet-modules-safetensors/resolve/87c3affbcad3baec52ffe39cac3a15a94902aed3/control_canny-fp16.safetensors"
    download-model --control-net "control_depth-fp16.safetensors" "https://huggingface.co./webui/ControlNet-modules-safetensors/resolve/87c3affbcad3baec52ffe39cac3a15a94902aed3/control_depth-fp16.safetensors"
    download-model --control-net "control_normal-fp16.safetensors" "https://huggingface.co./webui/ControlNet-modules-safetensors/resolve/87c3affbcad3baec52ffe39cac3a15a94902aed3/control_normal-fp16.safetensors"
    download-model --control-net "control_openpose-fp16.safetensors" "https://huggingface.co./webui/ControlNet-modules-safetensors/resolve/87c3affbcad3baec52ffe39cac3a15a94902aed3/control_openpose-fp16.safetensors"
    download-model --control-net "control_scribble-fp16.safetensors" "https://huggingface.co./webui/ControlNet-modules-safetensors/resolve/87c3affbcad3baec52ffe39cac3a15a94902aed3/control_scribble-fp16.safetensors"
    download-model --checkpoint "AtoZovyaRPGArtistTools15_sd15V1.safetensors" "https://civitai.com/api/download/models/10185"
    download-model --embedding "bad_prompt_version2.pt" "https://huggingface.co./datasets/Nerfgun3/bad_prompt/resolve/72fd9d6011c2ba87b5847b7e45e6603917e3cbed/bad_prompt_version2.pt"
    sed -i -e '/(modelmerger_interface, \"Checkpoint Merger\", \"modelmerger\"),/d' /app/stable-diffusion-webui/modules/ui.py
    sed -i -e '/(train_interface, \"Train\", \"ti\"),/d' /app/stable-diffusion-webui/modules/ui.py
    sed -i -e '/extensions_interface, \"Extensions\", \"extensions\"/d' /app/stable-diffusion-webui/modules/ui.py
    sed -i -e '/settings_interface, \"Settings\", \"settings\"/d' /app/stable-diffusion-webui/modules/ui.py
    rm -rf /app/stable-diffusion-webui/scripts /app/stable-diffusion-webui/extensions/deforum-for-automatic1111-webui /app/stable-diffusion-webui/extensions/stable-diffusion-webui-images-browser /app/stable-diffusion-webui/extensions/sd-civitai-browser /app/stable-diffusion-webui/extensions/sd-webui-additional-networks
    cp -f shared-config.json config.json
    cp -f shared-ui-config.json ui-config.json
    exit 0
fi
## End of lightweight installation for $IS_SHARED_UI setup.

## ----------------------------
## env $IS_SHARED_UI is not set
## ----------------------------

## Stable Diffusion 2.1 路 768 base model:
#download-model --checkpoint "v2-1_768-ema-pruned.safetensors" "https://huggingface.co./stabilityai/stable-diffusion-2-1/resolve/36a01dc742066de2e8c91e7cf0b8f6b53ef53da1/v2-1_768-ema-pruned.safetensors"
#download-model --checkpoint "v2-1_768-ema-pruned.yaml" "https://raw.githubusercontent.com/Stability-AI/stablediffusion/fc1488421a2761937b9d54784194157882cbc3b1/configs/stable-diffusion/v2-inference-v.yaml"

## Stable Diffusion 1.5 路 512 base model:
#download-model --checkpoint "v1-5-pruned-emaonly.safetensors" "https://huggingface.co./stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors"
#download-model --checkpoint "v1-5-pruned-emaonly.yaml" "https://huggingface.co./runwayml/stable-diffusion-v1-5/resolve/39593d5650112b4cc580433f6b0435385882d819/v1-inference.yaml"

## ----------------------------

## LoRA (low-rank adaptation) 路 epi_noiseoffset v2:
download-model --lora "hyperfusion100k_v4.safetensors" "https://civitai.com/api/download/models/19987"
download-model --lora "jimLeeDCComicsMarvel_offset.safetensors" "https://civitai.com/api/download/models/10580"
download-model --lora "epi_noiseoffset2.safetensors" "https://civitai.com/api/download/models/16576"
download-model --lora "agneseInnocente_1.safetensors" "https://civitai.com/api/download/models/34144"
download-model --lora "seethru_v10.safetensors" "https://civitai.com/api/download/models/32083"
download-model --lora "CommunityLoraExtract_lora320comicbabesV1.safetensors" "https://civitai.com/api/download/models/33744"
download-model --lora "CommunityLoraExtract_lora320revanimated.safetensors" "https://civitai.com/api/download/models/33787"

## ----------------------------

## VAE (variational autoencoder) 路 VAE 840k EMA:
download-model --vae "vae-ft-mse-840000-ema-pruned.safetensors" "https://huggingface.co./stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors"
download-model --vae "Grapefruit.vae.pt" "https://huggingface.co./iZELX1/Grapefruit/resolve/main/Grapefruit.vae.pt"
download-model --vae "kl-f8-anime.ckpt" "https://huggingface.co./hakurei/waifu-diffusion-v1-4/resolve/main/vae/kl-f8-anime.ckpt"

## ----------------------------

## ControlNet 路 Pre-extracted models:
download-model --control-net "cldm_v15.yaml" "https://huggingface.co./webui/ControlNet-modules-safetensors/resolve/87c3affbcad3baec52ffe39cac3a15a94902aed3/cldm_v15.yaml"
download-model --control-net "cldm_v21.yaml" "https://huggingface.co./webui/ControlNet-modules-safetensors/resolve/87c3affbcad3baec52ffe39cac3a15a94902aed3/cldm_v21.yaml"
download-model --control-net "control_canny-fp16.safetensors" "https://huggingface.co./webui/ControlNet-modules-safetensors/resolve/87c3affbcad3baec52ffe39cac3a15a94902aed3/control_canny-fp16.safetensors"
download-model --control-net "control_depth-fp16.safetensors" "https://huggingface.co./webui/ControlNet-modules-safetensors/resolve/87c3affbcad3baec52ffe39cac3a15a94902aed3/control_depth-fp16.safetensors"
download-model --control-net "control_hed-fp16.safetensors" "https://huggingface.co./webui/ControlNet-modules-safetensors/resolve/87c3affbcad3baec52ffe39cac3a15a94902aed3/control_hed-fp16.safetensors"
download-model --control-net "control_normal-fp16.safetensors" "https://huggingface.co./webui/ControlNet-modules-safetensors/resolve/87c3affbcad3baec52ffe39cac3a15a94902aed3/control_normal-fp16.safetensors"
download-model --control-net "control_openpose-fp16.safetensors" "https://huggingface.co./webui/ControlNet-modules-safetensors/resolve/87c3affbcad3baec52ffe39cac3a15a94902aed3/control_openpose-fp16.safetensors"
download-model --control-net "control_scribble-fp16.safetensors" "https://huggingface.co./webui/ControlNet-modules-safetensors/resolve/87c3affbcad3baec52ffe39cac3a15a94902aed3/control_scribble-fp16.safetensors"

## ----------------------------

## Embedding
download-model --embedding "bad_prompt_version2.pt" "https://huggingface.co./datasets/Nerfgun3/bad_prompt/resolve/72fd9d6011c2ba87b5847b7e45e6603917e3cbed/bad_prompt_version2.pt"
download-model --embedding "easynegative.safetensors" "https://huggingface.co./embed/EasyNegative/resolve/main/EasyNegative.safetensors"
download-model --embedding "microwaist_01bEmbedding.pt" "https://civitai.com/api/download/models/5246"

## ----------------------------

## Checkpoints:
download-model --checkpoint "babes_20.safetensors" "https://civitai.com/api/download/models/61372"
#download-model --checkpoint "animatrix_v11.safetensors" "https://civitai.com/api/download/models/34413"
download-model --checkpoint "XenoGasmArt.safetensors" "https://huggingface.co./Xenos14/zMine-TestModel/resolve/main/XenoGasmArt.safetensors"
#download-model --checkpoint "comicBabes_v1.safetensors" "https://civitai.com/api/download/models/24129"
download-model --checkpoint "XenoEngine5thDimension.safetensors" "https://huggingface.co./Xenos14/zMine-TestModel/resolve/main/XenoEngine5thDimension.safetensors"
download-model --checkpoint "galenaREDUX_v10.safetensors" "https://civitai.com/api/download/models/57724"
download-model --checkpoint "XenoEngine4.5.safetensors" "https://huggingface.co./Xenos14/zMine-TestModel/resolve/main/XenoEngine4.5.safetensors"


## ----------------------------

## Add additional models that you want to install on startup. Replace URL and FILENAME from the examples below with your values.

## Usage:
## download-model --checkpoint <filename> <url>
## download-model --lora <filename> <url>
## download-model --vae <filename> <url>
## download-model --control-net <filename> <url>
## download-model --embedding <filename> <url>

## ----------------------------


## Checkpoint 路 Example:
# download-model --checkpoint "FILENAME" "URL"

## LORA (low-rank adaptation) 路 Example:
# download-model --lora "FILENAME" "URL"

## VAE (variational autoencoder) 路 Example:
# download-model --vae "FILENAME" "URL"