浏览代码

Set PyTorch version to 2.0.1 for macOS

brkirch 2 年之前
父节点
当前提交
0cab07b2f1
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      webui-macos-env.sh

+ 1 - 1
webui-macos-env.sh

@@ -11,7 +11,7 @@ fi
 
 
 export install_dir="$HOME"
 export install_dir="$HOME"
 export COMMANDLINE_ARGS="--skip-torch-cuda-test --upcast-sampling --no-half-vae --use-cpu interrogate"
 export COMMANDLINE_ARGS="--skip-torch-cuda-test --upcast-sampling --no-half-vae --use-cpu interrogate"
-export TORCH_COMMAND="pip install torch torchvision"
+export TORCH_COMMAND="pip install torch==2.0.1 torchvision==0.15.2"
 export K_DIFFUSION_REPO="https://github.com/brkirch/k-diffusion.git"
 export K_DIFFUSION_REPO="https://github.com/brkirch/k-diffusion.git"
 export K_DIFFUSION_COMMIT_HASH="51c9778f269cedb55a4d88c79c0246d35bdadb71"
 export K_DIFFUSION_COMMIT_HASH="51c9778f269cedb55a4d88c79c0246d35bdadb71"
 export PYTORCH_ENABLE_MPS_FALLBACK=1
 export PYTORCH_ENABLE_MPS_FALLBACK=1