aboutsummaryrefslogtreecommitdiffstats
path: root/webui.sh
diff options
context:
space:
mode:
authorMuhammad Rizqi Nur <rizqinur2010@gmail.com>2022-11-19 08:20:07 +0000
committerMuhammad Rizqi Nur <rizqinur2010@gmail.com>2022-11-19 08:20:07 +0000
commitf1bdf2b15f32dffe9947fc14b107c1b8bd51ab55 (patch)
tree7d192cbb2f458427f41b2137eeeb68b2754537a2 /webui.sh
parentc8f7b5cdd73969d3d5027ceb71cbbd83d557702b (diff)
parentd9fd4525a5d684100997130cc4132736bab1e4d9 (diff)
downloadstable-diffusion-webui-gfx803-f1bdf2b15f32dffe9947fc14b107c1b8bd51ab55.tar.gz
stable-diffusion-webui-gfx803-f1bdf2b15f32dffe9947fc14b107c1b8bd51ab55.tar.bz2
stable-diffusion-webui-gfx803-f1bdf2b15f32dffe9947fc14b107c1b8bd51ab55.zip
Merge branch 'a1111' into vae-misc
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