|
#!/bin/bash |
|
|
|
|
|
CHECKPOINT=$1 |
|
|
|
|
|
RC=$2 |
|
|
|
GOORM="0" |
|
if [ -d "/workspace/firstContainer" ]; then |
|
GOORM="1" |
|
fi |
|
|
|
RELATIVE_DIR=`dirname "$0"` |
|
cd $RELATIVE_DIR |
|
|
|
if [ ! -d "SD" ]; then |
|
cd SD |
|
else |
|
cd SD* |
|
fi |
|
|
|
cd models |
|
cd Stable-diffusion |
|
|
|
CK_LINK="" |
|
|
|
case $CHECKPOINT in |
|
0) |
|
echo "CHECKPOINT pass." |
|
;; |
|
1) |
|
CK_LINK="https://huggingface.co/Magamanny/Pony-Diffusion-V6-XL/resolve/main/ponyDiffusionV6XL_v6StartWithThisOne.safetensors" |
|
;; |
|
2) |
|
CK_LINK="https://huggingface.co/JosefJilek/moeFussion/resolve/main/moeFussionV1.1.0_PDXL_Lx7-CLIP_VAE_FP16.safetensors" |
|
;; |
|
3) |
|
CK_LINK="https://huggingface.co/zuv0/test/resolve/main/MINTSDXL_LollipopMIX_A1-fp32.safetensors" |
|
;; |
|
4) |
|
CK_LINK="https://huggingface.co/Bulkbogan20/autismmix/resolve/main/autismmixSDXL_autismmixPony.safetensors" |
|
;; |
|
5) |
|
CK_LINK="https://huggingface.co/zuv0/test/resolve/main/lemontart_v20.safetensors" |
|
;; |
|
6) |
|
CK_LINK="https://huggingface.co/zuv0/test/resolve/main/grimoire_p0666.safetensors" |
|
;; |
|
a) |
|
CK_LINK="https://huggingface.co/cagliostrolab/animagine-xl-3.1/resolve/main/animagine-xl-3.1.safetensors" |
|
;; |
|
*) |
|
echo "CHECKPOINT error!" |
|
;; |
|
esac |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if [ -n "$CK_LINK" ]; then |
|
CK_FILE=${CK_LINK##*/} |
|
if [ -f "$CK_FILE" ]; then |
|
\mv -f "$CK_FILE" "../$CK_FILE" |
|
rm -rf *.safetensors |
|
rm -rf *.ckpt |
|
\mv -f "../$CK_FILE" "$CK_FILE" |
|
else |
|
rm -rf *.safetensors |
|
rm -rf *.ckpt |
|
if [ "$GOORM" == "1" ]; then |
|
wget –limit-rate=49999k -N "$CK_LINK" |
|
else |
|
wget -N "$CK_LINK" |
|
fi |
|
fi |
|
fi |
|
|
|
cd .. |
|
cd Lora |
|
|
|
case $RC in |
|
0) |
|
echo "RCXL pass." |
|
;; |
|
*) |
|
RCLIST="1 2 3 4 5 6 7 8 9" |
|
|
|
for var in $RCLIST |
|
do |
|
rm -rf RC$var |
|
done |
|
|
|
while [ $RC -gt 0 ]; do |
|
digit=$((RC % 10)) |
|
if [ ! -d "RCXL$digit" ]; then |
|
git clone https://huggingface.co/zuv0/RCXL$digit |
|
cd RCXL$digit |
|
git repack -a -d --depth=250 --window=250 |
|
cd .git |
|
rm -rf lfs |
|
cd .. |
|
cd .. |
|
fi |
|
RC=$((RC / 10)) |
|
done |
|
;; |
|
esac |
|
|