aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--requirements.txt1
-rw-r--r--webui-macos-env.sh6
2 files changed, 7 insertions, 0 deletions
diff --git a/requirements.txt b/requirements.txt
index 5f3d9623..05818aa6 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -30,3 +30,4 @@ inflection
GitPython
torchsde
safetensors
+psutil; sys_platform == 'darwin'
diff --git a/webui-macos-env.sh b/webui-macos-env.sh
index 68d1f754..95ca9c55 100644
--- a/webui-macos-env.sh
+++ b/webui-macos-env.sh
@@ -4,6 +4,12 @@
# Please modify webui-user.sh to change these instead of this file #
####################################################################
+if [[ -x "$(command -v python3.10)" ]]
+then
+ python_cmd="python3.10"
+fi
+
+export install_dir="$HOME"
export COMMANDLINE_ARGS="--skip-torch-cuda-test --no-half --use-cpu interrogate"
export TORCH_COMMAND="pip install torch==1.12.1 torchvision==0.13.1"
export K_DIFFUSION_REPO="https://github.com/brkirch/k-diffusion.git"