| #!/bin/bash
|
|
|
| FLAG_1=$1
|
| FLAG_2=$2
|
| FLAG_3=$3
|
| R_PATH="$(pwd)"
|
| CURRENT_DATE=$(date +"%Y%m%d_%H%M%S")
|
| MUID=""
|
| OUPUT_PATH=""
|
|
|
| if [ -d "$R_PATH/output" ]; then
|
| OUPUT_PATH="output"
|
| elif [ -d "$R_PATH/outputs" ]; then
|
| OUPUT_PATH="outputs"
|
| elif [ -d "$R_PATH/SD/output" ]; then
|
| OUPUT_PATH="SD/output"
|
| elif [ -d "$R_PATH/SD/outputs" ]; then
|
| OUPUT_PATH="SD/outputs"
|
| else
|
| OUPUT_PATH="SD/output"
|
| fi
|
|
|
| if [ -f "$R_PATH/SD/MUID.txt" ]; then
|
| MUID=$(<./SD/MUID.txt)
|
| MUID=${MUID%%+( )}
|
| MUID=${MUID%%+(\n)}
|
|
|
|
|
| else
|
| RANDOM_STR1="$(head -c 128 /dev/urandom | tr -dc 'A-Z')ZZ"
|
| RANDOM_STR2="$(head -c 384 /dev/urandom | tr -dc 'A-Z0-9')ZZZZZZZZZZZZZZ"
|
| MUID="${RANDOM_STR1::1}${RANDOM_STR2::14}${RANDOM_STR1:1:1}"
|
| echo -e "$MUID" >> ./SD/MUID.txt
|
| fi
|
|
|
| MUIDL="${MUID::3}${MUID:13:3}"
|
|
|
| if [[ "$FLAG_1" =~ h ]]; then
|
| echo -e " k : kill launch.sh(webui)\n @ : outputs/t2i folder to rename\n u : autoUpload MUID3+3\n f : forcePull\n o : out.sh\n g : gc.sh\n e : current chackpoint del\n d 0~9 : chackpoint download\n r : run.sh\n example : . e kdr 4\n y : runpodctl send\n"
|
| else
|
| if [[ "$FLAG_1" =~ k ]]; then
|
| kill -9 $(ps -ef | grep 'webui' | grep -v 'grep' | awk '{print $2}')
|
| kill -9 $(ps -ef | grep 'launch.py' | grep -v 'grep' | awk '{print $2}')
|
| kill -9 $(ps -ef | grep 'run.sh' | grep -v 'grep' | awk '{print $2}')
|
| fi
|
|
|
| if [[ "$FLAG_1" =~ @ ]]; then
|
| if [ -d "$R_PATH/$OUPUT_PATH/txt2img-images" ]; then
|
| RANDOM_STR="$(head -c 128 /dev/urandom | tr -dc 'A-Z')ZZZ"
|
| RENAME_NAME="t2i_${RANDOM_STR::4}${CURRENT_DATE}"
|
| \mv -f "$R_PATH/$OUPUT_PATH/txt2img-images" "$R_PATH/$OUPUT_PATH/${RENAME_NAME}"
|
| echo -e "rename txt2img-images -> ${RENAME_NAME}"
|
| fi
|
| fi
|
|
|
| if [[ "$FLAG_1" =~ u ]]; then
|
| curl -L -O https://huggingface.co/zuv0/SDXLFNC221/resolve/main/autoUpload.sh
|
| sleep 1
|
| bash -c ". ./autoUpload.sh $MUIDL"
|
| fi
|
|
|
| if [[ "$FLAG_1" =~ f ]]; then
|
| bash -c ". $R_PATH/forcePull.sh"
|
| fi
|
|
|
| if [[ "$FLAG_1" =~ o ]]; then
|
| bash -c ". $R_PATH/out.sh"
|
| fi
|
|
|
| if [[ "$FLAG_1" =~ g ]]; then
|
| bash -c ". $R_PATH/gc.sh"
|
| fi
|
|
|
| if [[ "$FLAG_1" =~ e ]]; then
|
| bash -c ". $R_PATH/SD/downs.sh gc"
|
| fi
|
|
|
| if [[ "$FLAG_1" =~ d ]]; then
|
| bash -c ". $R_PATH/SD/downs.sh $FLAG_2 0"
|
| fi
|
|
|
| if [[ "$FLAG_1" =~ r ]]; then
|
| bash -c ". $R_PATH/run.sh"
|
| fi
|
|
|
| if [[ "$FLAG_1" =~ y ]]; then
|
| bash -c "runpodctl send /workspace/$OUPUT_PATH"
|
| fi
|
|
|
| fi
|
|
|
