aboutsummaryrefslogtreecommitdiffstats
path: root/webui.sh
diff options
context:
space:
mode:
authorMuhammad Rizqi Nur <rizqinur2010@gmail.com>2022-11-19 08:21:00 +0000
committerMuhammad Rizqi Nur <rizqinur2010@gmail.com>2022-11-19 08:21:00 +0000
commit45dca0562e0a68d2f92a5b6fe0412dcd8ba0659e (patch)
treefe4131dd0f3251022c62120a5ef6bf79fa1242a3 /webui.sh
parent028b67b6357b5a00ccbd6ea72d2f244a6664162b (diff)
parentd9fd4525a5d684100997130cc4132736bab1e4d9 (diff)
downloadstable-diffusion-webui-gfx803-45dca0562e0a68d2f92a5b6fe0412dcd8ba0659e.tar.gz
stable-diffusion-webui-gfx803-45dca0562e0a68d2f92a5b6fe0412dcd8ba0659e.tar.bz2
stable-diffusion-webui-gfx803-45dca0562e0a68d2f92a5b6fe0412dcd8ba0659e.zip
Merge branch 'a1111' into vae-fix-none
Diffstat (limited to 'webui.sh')
-rwxr-xr-xwebui.sh16
1 files changed, 12 insertions, 4 deletions
diff --git a/webui.sh b/webui.sh
index cc06f344..d89b344b 100755
--- a/webui.sh
+++ b/webui.sh
@@ -134,7 +134,15 @@ else
exit 1
fi
-printf "\n%s\n" "${delimiter}"
-printf "Launching launch.py..."
-printf "\n%s\n" "${delimiter}"
-"${python_cmd}" "${LAUNCH_SCRIPT}" "$@"
+if [[ ! -z "${ACCELERATE}" ]] && [ ${ACCELERATE}="True" ] && [ -x "$(command -v accelerate)" ]
+then
+ printf "\n%s\n" "${delimiter}"
+ printf "Accelerating launch.py..."
+ printf "\n%s\n" "${delimiter}"
+ 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}"
+ "${python_cmd}" "${LAUNCH_SCRIPT}" "$@"
+fi