mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-01-31 04:42:09 -08:00
Code Issues Projects Releases Packages Wiki Activity

Supported Pytorch added for Maxwell.

Browse source
This commit is contained in:
Kavya Mali 2025-11-07 15:03:37 +05:30 • committed by GitHub
parent 6685e532df
commit 8544f3af4b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 12 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

13
modules/launch_utils.py
View file

@ -347,9 +347,19 @@ def early_access_blackwell_wheels():
return f'pip install {ea_whl.get(sys.version_info.minor)}'
def maxwell():
"""For Maxwell GPUs"""
cc = get_cuda_comp_cap()
if 5.0 <= cc < 6.0: # Maxwell, Compute Capability 5.0/5.2
torch_command = os.environ.get('TORCH_COMMAND', f"pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126")
return torch_command
return None
def prepare_environment():
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.0 torchvision==0.22.0 --extra-index-url {torch_index_url}")
torch_command = maxwell() or os.environ.get('TORCH_COMMAND', f"pip install torch==2.7.0 torchvision==0.22.0 --extra-index-url {torch_index_url}")
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
@ -512,3 +522,4 @@ def dump_sysinfo():
file.write(text)
return filename