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