update reqs for torch 2.7.1

This commit is contained in:
maybleMyers 2025-10-11 13:11:30 -07:00
parent 3dc3ee9bd6
commit 31e98a6f90
2 changed files with 3 additions and 3 deletions

View File

@ -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

View File

@ -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