From 31e98a6f90f0fac12e424d86f6ebee395b071f5f Mon Sep 17 00:00:00 2001 From: maybleMyers Date: Sat, 11 Oct 2025 13:11:30 -0700 Subject: [PATCH] update reqs for torch 2.7.1 --- modules/launch_utils.py | 4 ++-- requirements_versions.txt | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/modules/launch_utils.py b/modules/launch_utils.py index 420d9b19..59ff5e1b 100644 --- a/modules/launch_utils.py +++ b/modules/launch_utils.py @@ -361,8 +361,8 @@ def requirements_met(requirements_file): def prepare_environment(): - torch_index_url = os.environ.get('TORCH_INDEX_URL', "https://download.pytorch.org/whl/cu121") - torch_command = os.environ.get('TORCH_COMMAND', f"pip install torch==2.3.1 torchvision==0.18.1 --extra-index-url {torch_index_url}") + torch_index_url = os.environ.get('TORCH_INDEX_URL', "https://download.pytorch.org/whl/cu128") + torch_command = os.environ.get('TORCH_COMMAND', f"pip install torch==2.7.1+cu128 torchvision --index-url https://download.pytorch.org/whl/cu128") if args.use_ipex: if platform.system() == "Windows": # The "Nuullll/intel-extension-for-pytorch" wheels were built from IPEX source for Intel Arc GPU: https://github.com/intel/intel-extension-for-pytorch/tree/xpu-main diff --git a/requirements_versions.txt b/requirements_versions.txt index a6026209..bca09346 100644 --- a/requirements_versions.txt +++ b/requirements_versions.txt @@ -27,7 +27,7 @@ scikit-image==0.21.0 spandrel==0.3.4 spandrel-extra-arches==0.1.1 tomesd==0.1.3 -torch +torch==2.7.1 torchdiffeq==0.2.3 torchsde==0.2.6 transformers==4.46.1