From 3c0ca2e815493603af75706e1663ca049c730e50 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Thu, 13 Mar 2025 22:31:47 +0300 Subject: [PATCH] Add --upgrade to nightly torch commands --- CHANGELOG.md | 4 +++- installer.py | 8 ++++---- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index e9c2c69ac..6ceb44399 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,11 +1,13 @@ # Change Log for SD.Next -## Update for 2025-03-12 +## Update for 2025-03-13 - fix installer not starting when older version of rich is installed - fix circular imports when debug flags are enabled - fix cuda errors with directml - fix memory stats not displaying the ram usage +- fix remote vae not being stored in metadata, thanks @iDeNoh +- add --upgrade to torch_command when using --use-nightly for ipex and rocm - **ipex** - add xpu to profiler - fix untyped_storage, torch.eye and torch.cuda.device ops diff --git a/installer.py b/installer.py index 30e0cb377..153dae193 100644 --- a/installer.py +++ b/installer.py @@ -667,11 +667,11 @@ def install_rocm_zluda(): if args.use_nightly: if rocm.version is None or float(rocm.version) >= 6.3: # assume the latest if version check fails - torch_command = os.environ.get('TORCH_COMMAND', '--pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.3') + torch_command = os.environ.get('TORCH_COMMAND', '--upgrade --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.3') elif rocm.version == "6.2": # use rocm 6.2.4 instead of 6.2 as torch+rocm6.2 doesn't exists - torch_command = os.environ.get('TORCH_COMMAND', '--pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4') + torch_command = os.environ.get('TORCH_COMMAND', '--upgrade --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4') else: # oldest rocm version on nightly is 6.1 - torch_command = os.environ.get('TORCH_COMMAND', '--pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.1') + torch_command = os.environ.get('TORCH_COMMAND', '--upgrade --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.1') else: if rocm.version is None or float(rocm.version) >= 6.2: # assume the latest if version check fails # use rocm 6.2.4 instead of 6.2 as torch==2.6.0+rocm6.2 doesn't exists @@ -735,7 +735,7 @@ def install_ipex(torch_command): # os.environ.setdefault('IGC_EnableDPEmulation', '1') # FP64 Emulation if args.use_nightly: - torch_command = os.environ.get('TORCH_COMMAND', '--pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/xpu') + torch_command = os.environ.get('TORCH_COMMAND', '--upgrade --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/xpu') else: torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.6.0+xpu torchvision==0.21.0+xpu --index-url https://download.pytorch.org/whl/xpu')