mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
98fe49cb55
'requirements.in':
- add picklescan
- finally find a good compromise for torch (==1.12.0) and
torchvision (==0.13.0) across all platforms
'invoke.sh: hotfix for MacOS - add `export PYTORCH_ENABLE_MPS_FALLBACK=1`
Signed-off-by: Ben Alkov <ben.alkov@gmail.com>
31 lines
960 B
Bash
Executable File
31 lines
960 B
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
set -euo pipefail
|
|
IFS=$'\n\t'
|
|
|
|
source .venv/bin/activate
|
|
|
|
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 scripts/invoke.py;;
|
|
2 ) printf "\nStarting the InvokeAI browser-based UI..\n"; .venv/bin/python scripts/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
|
|
echo "'python' command is:"
|
|
which python
|
|
echo "'python' version is:"
|
|
python --version
|
|
echo "Type 'exit' to quit this shell"
|
|
fi
|