Spaces:
Runtime error
Runtime error
File size: 2,434 Bytes
c19ca42 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 |
#!/usr/bin/env bash
# -------------------------------------------------------------------------------------------------------------
# Do not make any changes to this file, change the variables in webui-user.sh instead and call this file
# -------------------------------------------------------------------------------------------------------------
# change to local directory
cd -- "$(dirname -- "$0")"
can_run_as_root=0
export ERROR_REPORTING=FALSE
export PIP_IGNORE_INSTALLED=0
# Read variables from webui-user.sh
if [[ -f webui-user.sh ]]
then
source ./webui-user.sh
fi
# python3 executable
if [[ -z "${PYTHON}" ]]
then
PYTHON="python3"
fi
# git executable
if [[ -z "${GIT}" ]]
then
export GIT="git"
fi
if [[ -z "${venv_dir}" ]]
then
venv_dir="venv"
fi
# read any command line flags to the webui.sh script
while getopts "f" flag > /dev/null 2>&1
do
case ${flag} in
f) can_run_as_root=1;;
*) break;;
esac
done
# Do not run as root unless inside a Docker container
if [[ $(id -u) -eq 0 && can_run_as_root -eq 0 && ! -f /.dockerenv ]]
then
echo "Cannot run as root"
exit 1
fi
for preq in "${GIT}" "${PYTHON}"
do
if ! hash "${preq}" &>/dev/null
then
printf "Error: %s is not installed, aborting...\n" "${preq}"
exit 1
fi
done
if ! "${PYTHON}" -c "import venv" &>/dev/null
then
echo "Error: python3-venv is not installed"
exit 1
fi
echo "Create and activate python venv"
if [[ ! -d "${venv_dir}" ]]
then
"${PYTHON}" -m venv "${venv_dir}"
first_launch=1
fi
if [[ -f "${venv_dir}"/bin/activate ]]
then
source "${venv_dir}"/bin/activate
else
echo "Error: Cannot activate python venv"
exit 1
fi
if [ -d "$(realpath "$venv_dir")/lib/" ] && [[ -z "${DISABLE_VENV_LIBS}" ]]
then
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$(realpath "$venv_dir")/lib/
fi
if [[ ! -z "${ACCELERATE}" ]] && [ ${ACCELERATE}="True" ] && [ -x "$(command -v accelerate)" ]
then
echo "Launching accelerate launch.py..."
exec accelerate launch --num_cpu_threads_per_process=6 launch.py "$@"
elif [[ "$@" == *"--use-ipex"* ]] && [[ -z "${first_launch}" ]] && [ -x "$(command -v ipexrun)" ] && [[ -z "${DISABLE_IPEXRUN}" ]]
then
echo "Launching ipexrun launch.py..."
exec ipexrun --multi-task-manager 'taskset' --memory-allocator 'jemalloc' launch.py "$@"
else
echo "Launching launch.py..."
exec "${PYTHON}" launch.py "$@"
fi
|