stable-diffusion-webui/webui.sh

187 lines
5.2 KiB
Bash
Raw Normal View History

#!/usr/bin/env bash
2022-09-13 13:28:04 +00:00
#################################################
# Please do not make any changes to this file, #
# change the variables in webui-user.sh instead #
#################################################
2022-10-07 19:25:01 +00:00
# If run from macOS, load defaults from webui-macos-env.sh
if [[ "$OSTYPE" == "darwin"* ]]; then
2023-03-12 17:50:02 +00:00
if [[ -f webui-macos-env.sh ]]
then
2023-03-12 17:50:02 +00:00
source ./webui-macos-env.sh
fi
fi
2022-09-13 13:28:04 +00:00
# Read variables from webui-user.sh
# shellcheck source=/dev/null
2023-03-12 17:50:02 +00:00
if [[ -f webui-user.sh ]]
2022-09-13 13:28:04 +00:00
then
2023-03-12 17:50:02 +00:00
source ./webui-user.sh
2022-09-13 13:28:04 +00:00
fi
2023-03-12 17:50:02 +00:00
2022-09-13 13:28:04 +00:00
# Set defaults
# Install directory without trailing slash
if [[ -z "${install_dir}" ]]
then
install_dir="/home/$(whoami)"
fi
# Name of the subdirectory (defaults to stable-diffusion-webui)
if [[ -z "${clone_dir}" ]]
then
clone_dir="stable-diffusion-webui"
fi
# python3 executable
if [[ -z "${python_cmd}" ]]
then
python_cmd="python3"
fi
# git executable
2022-09-13 15:28:54 +00:00
if [[ -z "${GIT}" ]]
2022-09-13 13:28:04 +00:00
then
2022-09-13 15:28:54 +00:00
export GIT="git"
2022-09-13 13:28:04 +00:00
fi
# python3 venv without trailing slash (defaults to ${install_dir}/${clone_dir}/venv)
if [[ -z "${venv_dir}" ]]
then
2023-03-12 17:50:02 +00:00
venv_dir="venv"
2022-09-13 13:28:04 +00:00
fi
2022-09-30 09:44:08 +00:00
if [[ -z "${LAUNCH_SCRIPT}" ]]
2022-09-30 08:51:05 +00:00
then
2023-03-12 17:50:02 +00:00
LAUNCH_SCRIPT="launch.py"
2022-09-30 08:51:05 +00:00
fi
2022-10-07 19:25:01 +00:00
# this script cannot be run as root by default
can_run_as_root=0
# read any command line flags to the webui.sh script
while getopts "f" flag > /dev/null 2>&1
2022-10-07 19:25:01 +00:00
do
case ${flag} in
f) can_run_as_root=1;;
*) break;;
2022-10-07 19:25:01 +00:00
esac
done
2022-09-26 10:55:10 +00:00
# Disable sentry logging
export ERROR_REPORTING=FALSE
2022-09-13 13:28:04 +00:00
# Do not reinstall existing pip packages on Debian/Ubuntu
export PIP_IGNORE_INSTALLED=0
# Pretty print
delimiter="################################################################"
printf "\n%s\n" "${delimiter}"
printf "\e[1m\e[32mInstall script for stable-diffusion + Web UI\n"
printf "\e[1m\e[34mTested on Debian 11 (Bullseye)\e[0m"
printf "\n%s\n" "${delimiter}"
# Do not run as root
2022-10-07 19:25:01 +00:00
if [[ $(id -u) -eq 0 && can_run_as_root -eq 0 ]]
2022-09-13 13:28:04 +00:00
then
printf "\n%s\n" "${delimiter}"
printf "\e[1m\e[31mERROR: This script must not be launched as root, aborting...\e[0m"
printf "\n%s\n" "${delimiter}"
exit 1
else
printf "\n%s\n" "${delimiter}"
printf "Running on \e[1m\e[32m%s\e[0m user" "$(whoami)"
printf "\n%s\n" "${delimiter}"
fi
if [[ -d .git ]]
then
printf "\n%s\n" "${delimiter}"
printf "Repo already cloned, using it as install directory"
printf "\n%s\n" "${delimiter}"
install_dir="${PWD}/../"
clone_dir="${PWD##*/}"
fi
# Check prerequisites
2023-01-19 18:21:02 +00:00
gpu_info=$(lspci 2>/dev/null | grep VGA)
case "$gpu_info" in
*"Navi 1"*|*"Navi 2"*) export HSA_OVERRIDE_GFX_VERSION=10.3.0
;;
*"Renoir"*) export HSA_OVERRIDE_GFX_VERSION=9.0.0
2023-01-19 23:21:05 +00:00
printf "\n%s\n" "${delimiter}"
2023-01-20 14:31:27 +00:00
printf "Experimental support for Renoir: make sure to have at least 4GB of VRAM and 10GB of RAM or enable cpu mode: --use-cpu all --no-half"
2023-01-19 23:21:51 +00:00
printf "\n%s\n" "${delimiter}"
;;
*)
;;
esac
2023-01-19 18:23:40 +00:00
if echo "$gpu_info" | grep -q "AMD" && [[ -z "${TORCH_COMMAND}" ]]
then
export TORCH_COMMAND="pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.2"
fi
2023-01-19 18:21:02 +00:00
for preq in "${GIT}" "${python_cmd}"
2022-09-13 13:28:04 +00:00
do
if ! hash "${preq}" &>/dev/null
then
printf "\n%s\n" "${delimiter}"
printf "\e[1m\e[31mERROR: %s is not installed, aborting...\e[0m" "${preq}"
printf "\n%s\n" "${delimiter}"
exit 1
fi
done
if ! "${python_cmd}" -c "import venv" &>/dev/null
then
printf "\n%s\n" "${delimiter}"
printf "\e[1m\e[31mERROR: python3-venv is not installed, aborting...\e[0m"
printf "\n%s\n" "${delimiter}"
exit 1
fi
2023-03-12 17:50:02 +00:00
cd "${install_dir}"/ || { printf "\e[1m\e[31mERROR: Can't cd to %s/, aborting...\e[0m" "${install_dir}"; exit 1; }
if [[ -d "${clone_dir}" ]]
2022-09-13 13:28:04 +00:00
then
2023-03-12 17:50:02 +00:00
cd "${clone_dir}"/ || { printf "\e[1m\e[31mERROR: Can't cd to %s/%s/, aborting...\e[0m" "${install_dir}" "${clone_dir}"; exit 1; }
else
2022-10-24 14:13:25 +00:00
printf "\n%s\n" "${delimiter}"
printf "Clone stable-diffusion-webui"
printf "\n%s\n" "${delimiter}"
2023-03-12 17:50:02 +00:00
"${GIT}" clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git "${clone_dir}"
cd "${clone_dir}"/ || { printf "\e[1m\e[31mERROR: Can't cd to %s/%s/, aborting...\e[0m" "${install_dir}" "${clone_dir}"; exit 1; }
2022-09-13 13:28:04 +00:00
fi
printf "\n%s\n" "${delimiter}"
printf "Create and activate python venv"
printf "\n%s\n" "${delimiter}"
2023-03-12 17:50:02 +00:00
cd "${install_dir}"/"${clone_dir}"/ || { printf "\e[1m\e[31mERROR: Can't cd to %s/%s/, aborting...\e[0m" "${install_dir}" "${clone_dir}"; exit 1; }
2022-09-13 13:28:04 +00:00
if [[ ! -d "${venv_dir}" ]]
then
"${python_cmd}" -m venv "${venv_dir}"
first_launch=1
fi
# shellcheck source=/dev/null
2022-09-13 15:28:54 +00:00
if [[ -f "${venv_dir}"/bin/activate ]]
2022-09-13 13:28:04 +00:00
then
2022-09-13 15:28:54 +00:00
source "${venv_dir}"/bin/activate
2022-09-13 13:28:04 +00:00
else
printf "\n%s\n" "${delimiter}"
printf "\e[1m\e[31mERROR: Cannot activate python venv, aborting...\e[0m"
printf "\n%s\n" "${delimiter}"
exit 1
fi
2022-11-09 17:15:17 +00:00
if [[ ! -z "${ACCELERATE}" ]] && [ ${ACCELERATE}="True" ] && [ -x "$(command -v accelerate)" ]
then
printf "\n%s\n" "${delimiter}"
printf "Accelerating launch.py..."
printf "\n%s\n" "${delimiter}"
2023-01-04 22:58:16 +00:00
exec accelerate launch --num_cpu_threads_per_process=6 "${LAUNCH_SCRIPT}" "$@"
2022-11-09 17:15:17 +00:00
else
printf "\n%s\n" "${delimiter}"
printf "Launching launch.py..."
2023-01-19 18:23:40 +00:00
printf "\n%s\n" "${delimiter}"
2023-01-19 18:21:02 +00:00
exec "${python_cmd}" "${LAUNCH_SCRIPT}" "$@"
2022-11-09 17:15:17 +00:00
fi