Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

add back PYTORCH_ENABLE_MPS_FALLBACK

Browse Source
This commit is contained in:
Lincoln Stein 2022-12-02 09:28:17 -05:00
parent 34d6f35408
commit 5da42e0ad2
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
binary_installer/invoke.sh.in
View File

@ -4,6 +4,11 @@ set -eu
. .venv/bin/activate
# set required env var for torch on mac MPS
if [ "$(uname -s)" == "Darwin" ]; then
export PYTORCH_ENABLE_MPS_FALLBACK=1
fi
echo "Do you want to generate images using the"
echo "1. command-line"
echo "2. browser-based UI"

5
source_installer/invoke.sh.in
View File

@ -10,6 +10,11 @@ source "$CONDA_BASEPATH/etc/profile.d/conda.sh" # otherwise conda complains abou
conda activate invokeai
# 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"