update torch to 2.1.0

This commit is contained in:
AUTOMATIC1111
2023-11-26 12:07:33 +03:00
parent 6955c210b7
commit 29f04149b6
3 changed files with 6 additions and 6 deletions

View File

@@ -11,7 +11,7 @@ fi
export install_dir="$HOME"
export COMMANDLINE_ARGS="--skip-torch-cuda-test --upcast-sampling --no-half-vae --use-cpu interrogate"
export TORCH_COMMAND="pip install torch==2.0.1 torchvision==0.15.2"
export TORCH_COMMAND="pip install torch==2.1.0 torchvision==0.16.0"
export PYTORCH_ENABLE_MPS_FALLBACK=1
####################################################################