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

fix(lora): case-insensitive stage detection for two-stage LoRA names

Browse Source
This commit is contained in:
CalamitousFelicitousness
2026-04-09 09:11:51 +01:00
parent a16c7ac16e
commit f1d2888f10
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/lora/extra_networks_lora.py
+4 -2
View File
@@ -131,9 +131,11 @@ def parse(p, params_list, step=0):
dyn_dims.append(dyn_dim)
lora_module = []
if 'high' in params.positional or 'HIGH 14B' in params.positional[0]:
name_lower = params.positional[0].lower()
positional_lower = [str(p).lower() for p in params.positional]
if 'high' in positional_lower or '_high' in name_lower or 'high ' in name_lower:
lora_module.append('transformer')
if 'low' in params.positional or 'LOW 14B' in params.positional[0]:
if 'low' in positional_lower or '_low' in name_lower or 'low ' in name_lower:
lora_module.append('transformer_2')
if params.named.get('module', None) is not None:
lora_module.append(params.named['module'].lower())