sdnext/webui.sh at master · vladmandic/sdnext · GitHub

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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
#!/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
use_uv=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
PYTHON_ENV="${PYTHON}"
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

for arg in "$@"
do
    if [[ "$arg" == "--uv" ]]
    then
        use_uv=1
        break
    fi
done

# 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 [[ "${use_uv}" -eq 1 ]]
then
    if ! hash "uv" &>/dev/null
    then
        echo "Warning: uv is not installed globally"
        use_uv=0
    fi
fi

if [[ "${use_uv}" -eq 0 ]]
then
    if ! "${PYTHON}" -c "import venv" &>/dev/null
    then
        echo "Error: python3-venv is not installed"
        exit 1
    fi
fi

if [[ ! -d "${venv_dir}" ]]
then
    if [[ "${use_uv}" -eq 1 ]]
    then
        echo "Create VENV: UV"
        uv venv "${venv_dir}"
    else
        echo "Create VENV: VENV"
        "${PYTHON}" -m venv "${venv_dir}"
    fi
    first_launch=1
fi

if [[ -f "${venv_dir}"/bin/activate ]]
then
    source "${venv_dir}"/bin/activate
    echo "Activate python venv: $VIRTUAL_ENV"
else
    echo "Error: Cannot activate python venv"
    exit 1
fi

# Add venv lib folder to PATH
if [ -d "$(realpath "$venv_dir")/lib/" ] && [[ -z "${DISABLE_VENV_LIBS}" ]]
then
    if [ -z "${LD_LIBRARY_PATH+x}" ]
    then
      export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$(realpath "$venv_dir")/lib/
    else
      export LD_LIBRARY_PATH=$(realpath "$venv_dir")/lib/
    fi
fi

# Add ROCm to PATH if it's not already
if  [ ! -x "$(command -v rocminfo)" ] && [ -f '/opt/rocm/bin/rocminfo' ]
then
    export PATH=$PATH:/opt/rocm/bin
    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/rocm/lib
fi

if [[ -n "${ACCELERATE}" ]] && [[ "${ACCELERATE}" == "True" ]] && [ -x "$(command -v accelerate)" ]
then
    echo "Launch: accelerate"
    exec accelerate launch --num_cpu_threads_per_process=6 launch.py "$@"
elif [[ -n "${IPEXRUN}" ]] && [[ "${IPEXRUN}" == "True" ]] && [ -x "$(command -v ipexrun)" ]
then
    echo "Launch: ipexrun"
    exec ipexrun --multi-task-manager 'taskset' --memory-allocator 'jemalloc' launch.py "$@"
elif [[ -f "${venv_dir}/bin/python3" ]]
then
    PYTHON="${venv_dir}/bin/python3"
    echo "Launch: ${PYTHON}"
    exec "${PYTHON}" launch.py "$@"
else
    echo "Launch: ${PYTHON}"
    exec "${PYTHON}" launch.py "$@"
fi