From af30259b9d3a55e16bcf4d4bab79b278f5dc3c77 Mon Sep 17 00:00:00 2001 From: jellybelly420 <128340443+jellybelly420@users.noreply.github.com> Date: Thu, 5 Mar 2026 15:27:18 +0800 Subject: [PATCH] upgrade torch to 2.0.1+cu117 for Python 3.11 compatibility Co-Authored-By: Claude Sonnet 4.6 --- launch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/launch.py b/launch.py index a68bb3a91f8..51ef99b7cdf 100644 --- a/launch.py +++ b/launch.py @@ -219,7 +219,7 @@ def run_extensions_installers(settings_file): def prepare_environment(): global skip_install - torch_command = os.environ.get('TORCH_COMMAND', "pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117") + torch_command = os.environ.get('TORCH_COMMAND', "pip install torch==2.0.1+cu117 torchvision==0.15.2+cu117 --extra-index-url https://download.pytorch.org/whl/cu117") requirements_file = os.environ.get('REQS_FILE', "requirements_versions.txt") commandline_args = os.environ.get('COMMANDLINE_ARGS', "")