File size: 12,321 Bytes
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
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
#!/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/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"

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

## LoRA (low-rank adaptation) 路 epi_noiseoffset v2:
#download-model --lora "epiNoiseoffset_v2.safetensors" "https://civitai.com/api/download/models/16576"

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

## 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/629b3ad3030ce36e15e70c5db7d91df0d60c627f/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 路 bad_prompt_version2
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"

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

## Checkpoints:
#download-model --checkpoint "revAnimated_v121.safetensors" "https://civitai.com/api/download/models/40248"
#download-model --checkpoint "megaMODEL2324CONTROLNETAND_24.safetensors" "https://civitai.com/api/download/models/39869?type=Model&format=SafeTensor"
#download-model --checkpoint "megaMODEL2324CONTROLNETAND_24.yaml" "https://civitai.com/api/download/models/39869?type=Config"
download-model --checkpoint "babes_11.safetensors" "https://civitai.com/api/download/models/21216"
#download-model --checkpoint "clarity_2.safetensors" "https://civitai.com/api/download/models/34070?type=Model&format=SafeTensor"
#download-model --checkpoint "animatrix_v11.safetensors" "https://civitai.com/api/download/models/34413"
#download-model --checkpoint "deliberate_v2.safetensors" "https://civitai.com/api/download/models/15236?type=Model&format=SafeTensor"
download-model --checkpoint "XenoEnginefp16.ckpt" "https://huggingface.co/Xenos14/zMine-TestModel/resolve/main/XenoEnginefp16.ckpt"
download-model --checkpoint "icomix_V04.safetensors" "https://civitai.com/api/download/models/43844"
download-model --checkpoint "galenaREDUX_v10.safetensors" "https://civitai.com/api/download/models/57724"
download-model --checkpoint "XenoEngine3.0.safetensors" "https://huggingface.co/Xenos14/zMine-TestModel/resolve/main/XenoEngine3.0.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>

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

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 "chianaFromFarscape_v10.safetensors" "https://civitai.com/api/download/models/29818"
download-model --lora "gta5ArtworkDiffusion_v1.safetensors" "https://civitai.com/api/download/models/8542"
download-model --lora "muscleGirl_v1.safetensors" "https://civitai.com/api/download/models/27749"
download-model --lora "incaseStyle_incaseAnythingV3.pt" "https://civitai.com/api/download/models/5938"
download-model --lora "aromaSenseiStyle_aromaSenseiAnything.pt" "https://civitai.com/api/download/models/6272"
download-model --lora "uberRealisticPornMerge_urpmv13296LORA.safetensors" "https://civitai.com/api/download/models/26552"
#download-model --lora "epiNoiseoffset_v2.safetensors" "https://civitai.com/api/download/models/16576"
download-model --lora "agneseInnocente_1.safetensors" "https://civitai.com/api/download/models/34144"
download-model --lora "kendraLust_v10.safetensors" "https://civitai.com/api/download/models/33262"
download-model --lora "seethru_v10.safetensors" "https://civitai.com/api/download/models/32083"
download-model --lora "honey_V01.safetensors" "https://civitai.com/api/download/models/33783"
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"

download-model --embedding "achechik.pt" "https://civitai.com/api/download/models/26280"
download-model --embedding "pam90s4nderson.pt" "https://civitai.com/api/download/models/28042"
download-model --embedding "sd15-keri-russell.pt" "https://civitai.com/api/download/models/29279"
download-model --embedding "microwaist_01bEmbedding.pt" "https://civitai.com/api/download/models/5246"


## 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"