diff --git a/installer.py b/installer.py index 32708ac6b..d726d3eb4 100644 --- a/installer.py +++ b/installer.py @@ -530,7 +530,7 @@ def check_torch(): else: if rocm_ver is None: # assume the latest if version check fails torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision --index-url https://download.pytorch.org/whl/rocm6.0') - elif rocm_ver == 6.1: # need nighlies + elif rocm_ver == "6.1": # need nightlies torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision --pre --index-url https://download.pytorch.org/whl/nightly/rocm6.1') elif int(rocm_ver.rsplit(".")[0]) <= 4 or (int(rocm_ver.rsplit(".")[0]) == 5 and int(rocm_ver.rsplit(".")[1]) <= 5): # oldest supported version is 5.5 log.warning(f"Unsupported ROCm version detected: {rocm_ver}") diff --git a/modules/hidiffusion/__init__.py b/modules/hidiffusion/__init__.py index 0e5b95772..44af02775 100644 --- a/modules/hidiffusion/__init__.py +++ b/modules/hidiffusion/__init__.py @@ -4,7 +4,8 @@ from modules.hidiffusion import hidiffusion def apply_hidiffusion(p): - hidiffusion.remove_hidiffusion(shared.sd_model) + if hasattr(shared.sd_model, "named_modules"): # Some pipelines doesn't have named_modules + hidiffusion.remove_hidiffusion(shared.sd_model) if p.hidiffusion: hidiffusion.is_aggressive_raunet = shared.opts.hidiffusion_steps > 0 hidiffusion.aggressive_step = shared.opts.hidiffusion_steps