diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-06-04 08:17:20 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-06-04 08:17:20 +0000 |
| commit | 0819383de05e57ec5da638bd4d5d180b5bac981a (patch) | |
| tree | ecec2c764772b337167248382fd7b72cdcef3209 /webui.sh | |
| parent | efc4c79b5ecd0bfff09b5f7bb9acf4b9044207ae (diff) | |
| parent | 333e63c0911c148ea306d7b72580d5c6d2f2c41a (diff) | |
| download | stable-diffusion-webui-gfx803-0819383de05e57ec5da638bd4d5d180b5bac981a.tar.gz stable-diffusion-webui-gfx803-0819383de05e57ec5da638bd4d5d180b5bac981a.tar.bz2 stable-diffusion-webui-gfx803-0819383de05e57ec5da638bd4d5d180b5bac981a.zip | |
Merge pull request #10975 from AUTOMATIC1111/restart3
A yet another method to restart webui.
Diffstat (limited to 'webui.sh')
| -rwxr-xr-x | webui.sh | 34 |
1 files changed, 20 insertions, 14 deletions
@@ -203,17 +203,23 @@ prepare_tcmalloc() { fi } -if [[ ! -z "${ACCELERATE}" ]] && [ ${ACCELERATE}="True" ] && [ -x "$(command -v accelerate)" ] -then - printf "\n%s\n" "${delimiter}" - printf "Accelerating launch.py..." - printf "\n%s\n" "${delimiter}" - prepare_tcmalloc - exec accelerate launch --num_cpu_threads_per_process=6 "${LAUNCH_SCRIPT}" "$@" -else - printf "\n%s\n" "${delimiter}" - printf "Launching launch.py..." - printf "\n%s\n" "${delimiter}" - prepare_tcmalloc - exec "${python_cmd}" "${LAUNCH_SCRIPT}" "$@" -fi +KEEP_GOING=1 +while [[ "$KEEP_GOING" -eq "1" ]]; do + if [[ ! -z "${ACCELERATE}" ]] && [ ${ACCELERATE}="True" ] && [ -x "$(command -v accelerate)" ]; then + printf "\n%s\n" "${delimiter}" + printf "Accelerating launch.py..." + printf "\n%s\n" "${delimiter}" + prepare_tcmalloc + accelerate launch --num_cpu_threads_per_process=6 "${LAUNCH_SCRIPT}" "$@" + else + printf "\n%s\n" "${delimiter}" + printf "Launching launch.py..." + printf "\n%s\n" "${delimiter}" + prepare_tcmalloc + "${python_cmd}" "${LAUNCH_SCRIPT}" "$@" + fi + + if [[ ! -f tmp/restart ]]; then + KEEP_GOING=0 + fi +done |