35 lines
1.1 KiB
Bash
35 lines
1.1 KiB
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
set -eu
|
||
|
|
|
||
|
|
# ensure we're in the correct folder in case user's CWD is somewhere else
|
||
|
|
scriptdir=$(dirname "$0")
|
||
|
|
cd "$scriptdir"
|
||
|
|
|
||
|
|
. .venv/bin/activate
|
||
|
|
|
||
|
|
export INVOKEAI_ROOT="$scriptdir"
|
||
|
|
|
||
|
|
# set required env var for torch on mac MPS
|
||
|
|
if [ "$(uname -s)" == "Darwin" ]; then
|
||
|
|
export PYTORCH_ENABLE_MPS_FALLBACK=1
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [ "$0" != "bash" ]; then
|
||
|
|
echo "Do you want to generate images using the"
|
||
|
|
echo "1. command-line"
|
||
|
|
echo "2. browser-based UI"
|
||
|
|
echo "3. open the developer console"
|
||
|
|
read -p "Please enter 1, 2, or 3: " yn
|
||
|
|
case $yn in
|
||
|
|
1 ) printf "\nStarting the InvokeAI command-line..\n"; .venv/bin/python .venv/bin/invoke.py $*;;
|
||
|
|
2 ) printf "\nStarting the InvokeAI browser-based UI..\n"; .venv/bin/python .venv/bin/invoke.py --web $*;;
|
||
|
|
3 ) printf "\nDeveloper Console:\n"; file_name=$(basename "${BASH_SOURCE[0]}"); bash --init-file "$file_name";;
|
||
|
|
* ) echo "Invalid selection"; exit;;
|
||
|
|
esac
|
||
|
|
else # in developer console
|
||
|
|
python --version
|
||
|
|
echo "Press ^D to exit"
|
||
|
|
export PS1="(InvokeAI) \u@\h \w> "
|
||
|
|
fi
|