mirror of
https://github.com/openvinotoolkit/stable-diffusion-webui.git
synced 2024-12-14 22:53:25 +03:00
install/launch scripts for linux
This commit is contained in:
orionaskatu
AUTOMATIC1111
parent
6bea45d495
commit
efc8ed13e1
32
README.md
32
README.md
@ -70,6 +70,36 @@ RealESRGAN into the directory with ESRGAN models. Thank you.
|
||||
- _*(optional)*_ place `GFPGANv1.3.pth` into webui directory, next to `webui.bat`.
|
||||
- run `webui-user.bat` from Windows Explorer. Run it as a normal user, ***not*** as administrator.
|
||||
|
||||
### Linux Automatic installation/launch
|
||||
|
||||
Prequisites:
|
||||
- For Debian-based:
|
||||
```commandline
|
||||
sudo apt install wget git python3 python3-venv
|
||||
```
|
||||
- For Red Hat-based:
|
||||
```commandline
|
||||
sudo dnf install wget git python3
|
||||
```
|
||||
|
||||
|
||||
- If you want to install to default directory `/home/$(whoami)/stable-diffusion-webui/`, you can launch directly:
|
||||
```commandline
|
||||
bash <(wget -qO- https://raw.githubusercontent.com/AUTOMATIC1111/stable-diffusion-webui/master/webui.sh)
|
||||
```
|
||||
|
||||
|
||||
- If you want to customize the installation just `git clone` the repo where you want it,
|
||||
change the variables in `webui-user.sh` and launch in console `bash webui.sh`.
|
||||
|
||||
|
||||
|
||||
- place `model.ckpt` into webui directory, next to `webui.py`.
|
||||
- _*(optional)*_ place `GFPGANv1.3.pth` into webui directory, next to `webui.py`.
|
||||
- run `bash webui.sh`. Run it as a normal user, ***not*** as root.
|
||||
|
||||
|
||||
|
||||
#### Troubleshooting
|
||||
|
||||
- if your version of Python is not in PATH (or if another version is), edit `webui-user.bat`, and modify the
|
||||
@ -330,4 +360,4 @@ After that follow the instructions in the `Manual instructions` section starting
|
||||
- Idea for SD upscale - https://github.com/jquesnelle/txt2imghd
|
||||
- CLIP interrogator idea and borrowing some code - https://github.com/pharmapsychotic/clip-interrogator
|
||||
- Initial Gradio script - posted on 4chan by an Anonymous user. Thank you Anonymous user.
|
||||
- (You)
|
||||
- (You)
|
||||
33
webui-user.sh
Normal file
33
webui-user.sh
Normal file
@ -0,0 +1,33 @@
|
||||
#!/bin/bash
|
||||
###########################################
|
||||
# Change the variables below to your need:#
|
||||
###########################################
|
||||
|
||||
# Install directory without trailing slash
|
||||
install_dir="/home/$(whoami)"
|
||||
|
||||
# Name of the subdirectory (defaults to stable-diffusion-webui)
|
||||
clone_dir="stable-diffusion-webui"
|
||||
|
||||
# Commandline arguments for webui.py, for example: commandline_args=(--medvram --opt-split-attention)
|
||||
commandline_args=()
|
||||
|
||||
# python3 executable
|
||||
python_cmd="python3"
|
||||
|
||||
# pip3 executable
|
||||
pip_cmd=(python3 -m pip)
|
||||
|
||||
# git executable
|
||||
git_cmd="git"
|
||||
|
||||
# python3 venv without trailing slash (defaults to ${install_dir}/${clone_dir}/venv)
|
||||
venv_dir="venv"
|
||||
|
||||
# pip3 install command for torch
|
||||
torch_command=(torch==1.12.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113)
|
||||
|
||||
# Requirements file to use for stable-diffusion-webui
|
||||
reqs_file="requirements_versions.txt"
|
||||
|
||||
###########################################
|
||||
220
webui.sh
Normal file
220
webui.sh
Normal file
@ -0,0 +1,220 @@
|
||||
#!/bin/bash
|
||||
#################################################
|
||||
# Please do not make any changes to this file, #
|
||||
# change the variables in webui-user.sh instead #
|
||||
#################################################
|
||||
# Read variables from webui-user.sh
|
||||
# shellcheck source=/dev/null
|
||||
if [[ -f webui-user.sh ]]
|
||||
then
|
||||
source ./webui-user.sh
|
||||
fi
|
||||
|
||||
# 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
|
||||
|
||||
# Commandline arguments for webui.py, for example: commandline_args=(--medvram --opt-split-attention)
|
||||
if [[ -z "${commandline_args}" ]]
|
||||
then
|
||||
commandline_args=()
|
||||
fi
|
||||
|
||||
# python3 executable
|
||||
if [[ -z "${python_cmd}" ]]
|
||||
then
|
||||
python_cmd="python3"
|
||||
fi
|
||||
|
||||
# pip3 executable
|
||||
if [[ -z "${pip_cmd}" ]]
|
||||
then
|
||||
pip_cmd=(python3 -m pip)
|
||||
fi
|
||||
|
||||
# git executable
|
||||
if [[ -z "${git_cmd}" ]]
|
||||
then
|
||||
git_cmd="git"
|
||||
fi
|
||||
|
||||
# python3 venv without trailing slash (defaults to ${install_dir}/${clone_dir}/venv)
|
||||
if [[ -z "${venv_dir}" ]]
|
||||
then
|
||||
venv_dir="venv"
|
||||
fi
|
||||
|
||||
# pip3 install command for torch
|
||||
if [[ -z "${torch_command}" ]]
|
||||
then
|
||||
torch_command=(torch==1.12.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113)
|
||||
fi
|
||||
|
||||
# Requirements file to use for stable-diffusion-webui
|
||||
if [[ -z "${reqs_file}" ]]
|
||||
then
|
||||
reqs_file="requirements_versions.txt"
|
||||
fi
|
||||
|
||||
# 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
|
||||
if [[ $(id -u) -eq 0 ]]
|
||||
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 prequisites
|
||||
for preq in git python3
|
||||
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
|
||||
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
printf "Clone or update stable-diffusion-webui"
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
cd "${install_dir}"/ || { printf "\e[1m\e[31mERROR: Can't cd to %s/, aborting...\e[0m" "${install_dir}"; exit 1; }
|
||||
if [[ -d "${clone_dir}" ]]
|
||||
then
|
||||
cd "${clone_dir}"/ || { printf "\e[1m\e[31mERROR: Can't cd to %s/%s/, aborting...\e[0m" "${install_dir}" "${clone_dir}"; exit 1; }
|
||||
"${git_cmd}" pull
|
||||
else
|
||||
"${git_cmd}" 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; }
|
||||
fi
|
||||
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
printf "Clone or update other repositories"
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
if [[ ! -d repositories ]]
|
||||
then
|
||||
mkdir repositories
|
||||
fi
|
||||
cd repositories || { printf "\e[1m\e[31mERROR: Can't cd to %s/%s/repositories/, aborting...\e[0m" "${install_dir}" "${clone_dir}"; exit 1; }
|
||||
|
||||
for repo in stable-diffusion taming-transformers CodeFormer BLIP
|
||||
do
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
printf "%s" "${repo}"
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
|
||||
if [[ -d "${repo}" ]]
|
||||
then
|
||||
cd "${repo}"/ || { printf "\e[1m\e[31mERROR: Can't cd to %s/stable-diffusion/repositories/%s, aborting...\e[0m" "${install_dir}" "${repo}"; exit 1; }
|
||||
"${git_cmd}" pull
|
||||
cd ..
|
||||
else
|
||||
if [[ "${repo}" == "stable-diffusion" || "${repo}" == "taming-transformers" ]]
|
||||
then
|
||||
"${git_cmd}" clone https://github.com/CompVis/"${repo}".git
|
||||
elif [[ "${repo}" == "CodeFormer" ]]
|
||||
then
|
||||
"${git_cmd}" clone https://github.com/sczhou/"${repo}".git
|
||||
elif [[ "${repo}" == "BLIP" ]]
|
||||
then
|
||||
"${git_cmd}" clone https://github.com/salesforce/"${repo}".git
|
||||
fi
|
||||
fi
|
||||
done
|
||||
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
printf "Create and activate python venv"
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
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; }
|
||||
if [[ ! -d "${venv_dir}" ]]
|
||||
then
|
||||
"${python_cmd}" -m venv "${venv_dir}"
|
||||
first_launch=1
|
||||
fi
|
||||
# shellcheck source=/dev/null
|
||||
if source "${venv_dir}"/bin/activate
|
||||
then
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
printf "Install dependencies"
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
"${pip_cmd[@]}" install "${torch_command[@]}"
|
||||
"${pip_cmd[@]}" install wheel transformers==4.19.2 diffusers invisible-watermark --prefer-binary
|
||||
"${pip_cmd[@]}" install git+https://github.com/crowsonkb/k-diffusion.git@1a0703dfb7d24d8806267c3e7ccc4caf67fd1331 --prefer-binary --only-binary=psutil
|
||||
"${pip_cmd[@]}" install git+https://github.com/TencentARC/GFPGAN.git@8d2447a2d918f8eba5a4a01463fd48e45126a379 --prefer-binary
|
||||
"${pip_cmd[@]}" install -r "${reqs_file}" --prefer-binary
|
||||
"${pip_cmd[@]}" install -r repositories/CodeFormer/requirements.txt --prefer-binary
|
||||
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
|
||||
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
printf "Check if models are present"
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
for model in GFPGANv1.3.pth model.ckpt
|
||||
do
|
||||
if [[ ! -f "${model}" ]]
|
||||
then
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
printf "\e[1m\e[33mWarning:\e[0m %s file not found..." "${model}"
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
if [[ "${model}" == "model.ckpt" ]] && [[ -n "${first_launch}" ]]
|
||||
then
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
printf "Place \e[1m\e[32m%s\e[0m into webui directory, next to \e[1m\e[32mwebui.py\e[0m\n" "${model}"
|
||||
printf "Then press a key to continue...\n"
|
||||
read -rsn 1
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
fi
|
||||
fi
|
||||
done
|
||||
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
printf "Launching webui.py..."
|
||||
printf "\n%s\n" "${delimiter}"
|
||||
"${python_cmd}" webui.py "${commandline_args[@]}"
|
||||
Loading…
Reference in New Issue
Block a user