Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

Cleanup

Browse Source
This commit is contained in:
Disty0
2024-05-17 18:57:29 +03:00
parent beae2f88a6
commit 2aadaa3923
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
installer.py
+1 -1
View File
@@ -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}")
modules/hidiffusion/__init__.py
+2 -1
View File
@@ -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