zuv0 commited on
Commit
0fa88c0
·
verified ·
1 Parent(s): cb04aa1

Upload 7 files

Browse files
Files changed (7) hide show
  1. autoUpload.sh +117 -0
  2. e +90 -0
  3. forcePull_rp.sh +11 -0
  4. gc.sh +23 -0
  5. install_rp.sh +90 -0
  6. out_rp.sh +49 -0
  7. run_rp.sh +19 -0
autoUpload.sh ADDED
@@ -0,0 +1,117 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+
3
+ # version 1
4
+
5
+ # output filename add
6
+ FLAG_NAME=$1
7
+
8
+ # null
9
+ FLAG_NULL=$2
10
+
11
+ # flag Key
12
+ FLAG_KEY=$3
13
+
14
+ if [ -z "$FLAG_NAME" ] || [ "$FLAG_NAME" = " " ]; then
15
+ FLAG_NAME="null"
16
+ fi
17
+ if [ -z "$FLAG_NULL" ]; then
18
+ FLAG_NULL="null"
19
+ fi
20
+ if [ -z "$FLAG_KEY" ]; then
21
+ FLAG_KEY="null"
22
+ fi
23
+
24
+ cd /workspace
25
+ cd *
26
+
27
+ R_PATH="$(pwd)"
28
+
29
+ if [ ! -d "$R_PATH/SD" ]; then
30
+ echo "SD folder not found."
31
+ exit 101
32
+ fi
33
+
34
+ MUID=""
35
+
36
+ if [ -f "$R_PATH/SD/MUID.txt" ]; then
37
+ MUID=$(<./SD/MUID.txt)
38
+ MUID=${MUID%%+( )}
39
+ MUID=${MUID%%+(\n)}
40
+ #MUID_LEN=${#MUID}
41
+ #MUID=${MUID:1:MUID_LEN-2}
42
+ else
43
+ RANDOM_STR1="$(head -c 128 /dev/urandom | tr -dc 'A-Z')ZZ"
44
+ RANDOM_STR2="$(head -c 384 /dev/urandom | tr -dc 'A-Z0-9')ZZZZZZZZZZZZZZ"
45
+ MUID="${RANDOM_STR1::1}${RANDOM_STR2::14}${RANDOM_STR1:1:1}"
46
+ echo -e "$MUID" >> ./SD/MUID.txt
47
+ fi
48
+
49
+ if [ ! -f "$R_PATH/SD/autoUpload_Log.txt" ]; then
50
+ touch ./SD/autoUpload_Log.txt
51
+ fi
52
+
53
+ if [ ! -f "$R_PATH/SD/autoUpload_UploadQueueIng.txt" ]; then
54
+ touch ./SD/autoUpload_UploadQueueIng.txt
55
+ fi
56
+
57
+ if [ ! -f "$R_PATH/SD/autoUpload_UploadQueueDone.txt" ]; then
58
+ touch ./SD/autoUpload_UploadQueueDone.txt
59
+ fi
60
+
61
+ OUPUT_PATH=""
62
+
63
+ if [ -d "$R_PATH/output" ]; then
64
+ OUPUT_PATH="output"
65
+ elif [ -d "$R_PATH/outputs" ]; then
66
+ OUPUT_PATH="outputs"
67
+ elif [ -d "$R_PATH/SD/output" ]; then
68
+ OUPUT_PATH="SD/output"
69
+ elif [ -d "$R_PATH/SD/outputs" ]; then
70
+ OUPUT_PATH="SD/outputs"
71
+ elif [ -d "$R_PATH/CU/output" ]; then
72
+ OUPUT_PATH="CU/output"
73
+ elif [ -d "$R_PATH/CU/outputs" ]; then
74
+ OUPUT_PATH="CU/outputs"
75
+ else
76
+ echo "output folder not found."
77
+ exit 102
78
+ fi
79
+
80
+ #echo "$FLAG_NAME $FLAG_NULL $FLAG_KEY $R_PATH $OUPUT_PATH $MUID"
81
+
82
+ if [ "$FLAG_KEY" = "null" ]; then
83
+ # enter when the user runs manually.
84
+ apt-get install -y inotify-tools
85
+ apt-get install -y zip
86
+ pip3 install pcloud
87
+ CURRNET_PID=$$
88
+ #kill -9 $(ps -ef | grep 'autoUpload' | grep -v 'grep' | awk '{print $2}')
89
+ kill -9 $(ps -ef | grep 'autoUpload' | grep -v 'grep' | awk -v pid=$CURRNET_PID '$2 != pid {print $2}')
90
+ FLAG_KEY="startDaemon"
91
+ bash -c ". $R_PATH/autoUpload.sh $FLAG_NAME $FLAG_NULL $FLAG_KEY"
92
+ elif [ "$FLAG_KEY" = "startDaemon" ]; then
93
+ cd /workspace
94
+ cd *
95
+ FLAG_KEY="runningDaemon"
96
+ chmod 755 autoUpload.sh
97
+ nohup ./autoUpload.sh $FLAG_NAME $FLAG_NULL $FLAG_KEY 2>&1 &
98
+ elif [ "$FLAG_KEY" = "runningDaemon" ]; then
99
+ cd /workspace
100
+ cd *
101
+ while :
102
+ do
103
+ bash -c ". $R_PATH/out.sh 1 $FLAG_NAME"
104
+ sleep 288s
105
+ for item in $(find . -name "output_${FLAG_NAME}*.zip"); do
106
+ rm -rf $item
107
+ done
108
+ sleep 10s
109
+ done
110
+ elif [ "$FLAG_KEY" = "clean" ]; then
111
+ rm -rf "$R_PATH/SD/autoUpload_UploadQueueIng.txt"
112
+ rm -rf "$R_PATH/SD/autoUpload_UploadQueueDone.txt"
113
+ elif [ "$FLAG_KEY" = "kill" ]; then
114
+ kill -15 $(ps -ef | grep 'autoUpload.sh' | grep -v 'grep' | awk '{print $2}')
115
+ else
116
+ :
117
+ fi
e ADDED
@@ -0,0 +1,90 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+
3
+ FLAG_1=$1
4
+ FLAG_2=$2
5
+ FLAG_3=$3
6
+ R_PATH="$(pwd)"
7
+ CURRENT_DATE=$(date +"%Y%m%d_%H%M%S")
8
+ MUID=""
9
+ OUPUT_PATH=""
10
+
11
+ if [ -d "$R_PATH/output" ]; then
12
+ OUPUT_PATH="output"
13
+ elif [ -d "$R_PATH/outputs" ]; then
14
+ OUPUT_PATH="outputs"
15
+ elif [ -d "$R_PATH/SD/output" ]; then
16
+ OUPUT_PATH="SD/output"
17
+ elif [ -d "$R_PATH/SD/outputs" ]; then
18
+ OUPUT_PATH="SD/outputs"
19
+ else
20
+ OUPUT_PATH="SD/output"
21
+ fi
22
+
23
+ if [ -f "$R_PATH/SD/MUID.txt" ]; then
24
+ MUID=$(<./SD/MUID.txt)
25
+ MUID=${MUID%%+( )}
26
+ MUID=${MUID%%+(\n)}
27
+ #MUID_LEN=${#MUID}
28
+ #MUID=${MUID:1:MUID_LEN-2}
29
+ else
30
+ RANDOM_STR1="$(head -c 128 /dev/urandom | tr -dc 'A-Z')ZZ"
31
+ RANDOM_STR2="$(head -c 384 /dev/urandom | tr -dc 'A-Z0-9')ZZZZZZZZZZZZZZ"
32
+ MUID="${RANDOM_STR1::1}${RANDOM_STR2::14}${RANDOM_STR1:1:1}"
33
+ echo -e "$MUID" >> ./SD/MUID.txt
34
+ fi
35
+
36
+ MUIDL="${MUID::3}${MUID:13:3}"
37
+
38
+ if [[ "$FLAG_1" =~ h ]]; then
39
+ 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"
40
+ else
41
+ if [[ "$FLAG_1" =~ k ]]; then
42
+ kill -9 $(ps -ef | grep 'webui' | grep -v 'grep' | awk '{print $2}')
43
+ kill -9 $(ps -ef | grep 'launch.py' | grep -v 'grep' | awk '{print $2}')
44
+ kill -9 $(ps -ef | grep 'run.sh' | grep -v 'grep' | awk '{print $2}')
45
+ fi
46
+
47
+ if [[ "$FLAG_1" =~ @ ]]; then
48
+ if [ -d "$R_PATH/$OUPUT_PATH/txt2img-images" ]; then
49
+ RANDOM_STR="$(head -c 128 /dev/urandom | tr -dc 'A-Z')ZZZ"
50
+ RENAME_NAME="t2i_${RANDOM_STR::4}${CURRENT_DATE}"
51
+ \mv -f "$R_PATH/$OUPUT_PATH/txt2img-images" "$R_PATH/$OUPUT_PATH/${RENAME_NAME}"
52
+ echo -e "rename txt2img-images -> ${RENAME_NAME}"
53
+ fi
54
+ fi
55
+
56
+ if [[ "$FLAG_1" =~ u ]]; then
57
+ curl -L -O https://huggingface.co/zuv0/SD15N2/resolve/main/autoUpload.sh
58
+ sleep 1
59
+ bash -c ". ./autoUpload.sh $MUIDL"
60
+ fi
61
+
62
+ if [[ "$FLAG_1" =~ f ]]; then
63
+ bash -c ". $R_PATH/forcePull.sh"
64
+ fi
65
+
66
+ if [[ "$FLAG_1" =~ o ]]; then
67
+ bash -c ". $R_PATH/out.sh"
68
+ fi
69
+
70
+ if [[ "$FLAG_1" =~ g ]]; then
71
+ bash -c ". $R_PATH/gc.sh"
72
+ fi
73
+
74
+ if [[ "$FLAG_1" =~ e ]]; then
75
+ bash -c ". $R_PATH/SD/downs.sh gc"
76
+ fi
77
+
78
+ if [[ "$FLAG_1" =~ d ]]; then
79
+ bash -c ". $R_PATH/SD/downs.sh $FLAG_2 0"
80
+ fi
81
+
82
+ if [[ "$FLAG_1" =~ r ]]; then
83
+ bash -c ". $R_PATH/run_rp.sh"
84
+ fi
85
+
86
+ if [[ "$FLAG_1" =~ y ]]; then
87
+ bash -c "runpodctl send /workspace/$OUPUT_PATH"
88
+ fi
89
+
90
+ fi
forcePull_rp.sh ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+
3
+ cd /workspace
4
+ git -C /workspace/SD/models fetch --all
5
+ git -C /workspace/SD/models reset --hard origin/main
6
+ git -C /workspace/SD/models pull
7
+ git -C /workspace/SD/models repack -a -d --depth=250 --window=250
8
+ rsync -r "/workspace/SD/models/embeddings/" "/workspace/SD/embeddings/"
9
+ rm -r "/workspace/SD/models/.git/lfs"
10
+
11
+ cd /workspace
gc.sh ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+
3
+ FLAG_1=$1
4
+
5
+ cd /workspace
6
+ rm -rf output
7
+ rm -rf outputs
8
+ rm -f out*.tar*
9
+ rm -f out*.zip*
10
+
11
+ cd SD*
12
+ rm -rf output
13
+ rm -rf outputs
14
+ rm -f out*.tar*
15
+ rm -f out*.zip*
16
+
17
+ if [ ${FLAG_1} -eq "1" ]; then
18
+ sudo rm -rf /tmp/*
19
+ else
20
+ find /tmp -mtime +2 -delete
21
+ fi
22
+
23
+ cd /workspace
install_rp.sh ADDED
@@ -0,0 +1,90 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+
3
+ cd /workspace
4
+ mkdir -p tmp
5
+ rm -rf README.md
6
+ rm -rf s
7
+ rm -rf ss.sh
8
+
9
+ \cp -f /workspace/SD/e /workspace/e
10
+ \cp -f /workspace/SD/run_rp.sh /workspace/run.sh
11
+ \cp -f /workspace/SD/out_rp.sh /workspace/out.sh
12
+ \cp -f /workspace/SD/forcePull_rp.sh /workspace/forcePull.sh
13
+ \cp -f /workspace/SD/gc.sh /workspace/gc.sh
14
+ \cp -f /workspace/SD/MyRP.ipynb /workspace/MyRP.ipynb
15
+
16
+ kill -9 $(ps -ef | grep 'relauncher.py' | grep -v 'grep' | awk '{print $2}')
17
+ kill -9 $(ps -ef | grep 'webui' | grep -v 'grep' | awk '{print $2}')
18
+ kill -9 $(ps -ef | grep 'launch.py' | grep -v 'grep' | awk '{print $2}')
19
+
20
+ rm -rf /workspace/stable-diffusion-webui
21
+
22
+ apt -y update -qq
23
+ apt-get install git-lfs
24
+ apt-get -y install rsync
25
+ apt-get -y install zip
26
+ #apt-get -y install aria2 unzip
27
+ """
28
+ curl -sS https://bootstrap.pypa.io/get-pip.py | python3.10
29
+ curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | sudo bash
30
+ """
31
+ git lfs install --force
32
+
33
+ cd /workspace/SD
34
+ curl -L -O https://huggingface.co/zuv0/test/resolve/main/SD15N2.tar
35
+ tar -xf SD15N2.tar
36
+ rm -rf SD15N2.tar
37
+ git clone https://huggingface.co/zu01ze/models15 models
38
+
39
+ cd /workspace/SD/models
40
+ git fetch --all
41
+ git reset --hard origin/main
42
+ git pull
43
+ rsync -r "embeddings/" "../embeddings/"
44
+ git repack -a -d --depth=250 --window=250
45
+ cd .git
46
+ rm -rf lfs
47
+
48
+ cd /workspace/SD
49
+ curl -L -O https://huggingface.co/zuv0/SD17/resolve/main/downs.sh
50
+ sh downs.sh $1 0
51
+
52
+ pip3 install pcloud
53
+ """
54
+ pip uninstall -y setuptools
55
+ pip install setuptools==69.5.1
56
+ pip uninstall -y matplotlib
57
+ pip install matplotlib==3.8.4
58
+ pip uninstall -y xformers
59
+ pip uninstall -y torchaudio
60
+ pip install torch==2.3.0 torchvision --index-url https://download.pytorch.org/whl/cu121
61
+ pip3 install -r requirements_versions.txt --extra-index-url https://download.pytorch.org/whl/cu121
62
+ pip3 install -r extensions-builtin/sd_forge_controlnet/requirements.txt
63
+ pip3 install -r extensions-builtin/forge_legacy_preprocessors/requirements.txt
64
+ #pip3 install insightface
65
+ pip3 uninstall -y onnxruntime
66
+ pip3 install onnxruntime-gpu
67
+ pip install pydantic==1.10.11
68
+ #apt install libgl1 -yq
69
+ apt-get install -y libgl1-mesa-glx
70
+ #apt -y install -qq aria2 libcairo2-dev pkg-config python3-dev
71
+ """
72
+ #python3 -c "from launch import prepare_environment; prepare_environment()" --skip-torch-cuda-test
73
+
74
+ cd /workspace
75
+
76
+ python3 /workspace/SD/launch.py --no-download-sd-model --exit
77
+
78
+ apt-get clean
79
+ pip cache purge
80
+
81
+ kill -9 $(ps -ef | grep 'relauncher.py' | grep -v 'grep' | awk '{print $2}')
82
+ kill -9 $(ps -ef | grep 'webui' | grep -v 'grep' | awk '{print $2}')
83
+ kill -9 $(ps -ef | grep 'launch.py' | grep -v 'grep' | awk '{print $2}')
84
+
85
+ rm -rf /workspace/stable-diffusion-webui
86
+ cd /workspace
87
+
88
+ echo " -----===== =====----- "
89
+ echo " Install Done."
90
+ echo " -----===== =====----- "
out_rp.sh ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+
3
+ # if == "1" running
4
+ OUT_CLOUD=$1
5
+
6
+ # file add
7
+ NAME=$2
8
+ if [ -z "$NAME" ]; then
9
+ NAME=""
10
+ else
11
+ NAME="${NAME}_"
12
+ fi
13
+
14
+ if command -v zip >/dev/null 2>&1; then
15
+ :
16
+ else
17
+ apt-get install zip -y
18
+ pip install pcloud
19
+ fi
20
+
21
+ CURRENT_DATE=$(date +"%Y%m%d_%H%M%S")
22
+ ZIP_FILENAME="output_${NAME}${CURRENT_DATE}.zip"
23
+ W_PATH="/workspace"
24
+ if [ -d "/workspace" ]; then
25
+ W_PATH="/workspace"
26
+ fi
27
+
28
+ cd "$W_PATH"
29
+
30
+ if [ -d "$W_PATH/SD/output" ]; then
31
+ echo "Creating archive. Please wait..."
32
+ zip -q -r "$W_PATH/$ZIP_FILENAME" "$W_PATH/SD/output"
33
+ echo "archive file : $ZIP_FILENAME"
34
+ fi
35
+
36
+ if [ -d "$W_PATH/output" ]; then
37
+ echo "Creating archive. Please wait..."
38
+ zip -q -r "$W_PATH/$ZIP_FILENAME" "$W_PATH/output"
39
+ echo "archive file : $ZIP_FILENAME"
40
+ fi
41
+
42
+ PD="'$(echo eDg3NjV4eHhAZ21haWwuY29t | base64 --decode)', 'Number'"
43
+
44
+ if [ -f "$W_PATH/$ZIP_FILENAME" ]; then
45
+ export ZIP_FILENAME
46
+ if [ "$OUT_CLOUD" = "1" ]; then
47
+ python3.10 -c "import os; from pcloud import PyCloud; pc = PyCloud($PD); pc.uploadfile(files=['$W_PATH/$ZIP_FILENAME', '$W_PATH/$ZIP_FILENAME'], path='/')"
48
+ fi
49
+ fi
run_rp.sh ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+
3
+ cd /workspace
4
+
5
+ # Spot Instance vs
6
+ if dpkg-query -W libgl1 2>/dev/null | grep -q "libgl1"; then
7
+ echo -n ""
8
+ else
9
+ echo "libgl1 is not installed. Installing..."
10
+ apt update
11
+ apt install -yq libgl1
12
+ fi
13
+
14
+ python /workspace/SD/launch.py --port 7860 --api --theme dark --allow-code --enable-insecure-extension-access --listen --share --always-high-vram --disable-safe-unpickle --cuda-malloc --cuda-stream --pin-shared-memory --enable-insecure-extension-access --ckpt-dir '/workspace/SD/models/Stable-diffusion' --hypernetwork-dir '/workspace/SD/models/hypernetworks' --embeddings-dir '/workspace/SD/embeddings' --lora-dir '/workspace/SD/models/Lora' --lyco-dir '/workspace/SD/models/Lora'
15
+
16
+ kill $!
17
+ kill -9 $(ps -ef | grep 'ngrok' | grep -v 'grep' | awk '{print $2}')
18
+ deactivate
19
+ echo "stopped"