mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-06-24 21:07:35 +08:00
update reqs for torch 2.7.1
This commit is contained in:
parent
3dc3ee9bd6
commit
31e98a6f90
@ -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
|
||||
|
||||
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user