webui-macos-env.sh 751 B

1234567891011121314151617
  1. #!/bin/bash
  2. ####################################################################
  3. # macOS defaults #
  4. # Please modify webui-user.sh to change these instead of this file #
  5. ####################################################################
  6. export install_dir="$HOME"
  7. export COMMANDLINE_ARGS="--skip-torch-cuda-test --upcast-sampling --no-half-vae --use-cpu interrogate"
  8. export PYTORCH_ENABLE_MPS_FALLBACK=1
  9. if [[ "$(sysctl -n machdep.cpu.brand_string)" =~ ^.*"Intel".*$ ]]; then
  10. export TORCH_COMMAND="pip install torch==2.1.2 torchvision==0.16.2"
  11. else
  12. export TORCH_COMMAND="pip install torch==2.3.1 torchvision==0.18.1"
  13. fi
  14. ####################################################################