From f1d2888f10e16b070bc97ee4cab2fd87ba5b44a7 Mon Sep 17 00:00:00 2001 From: CalamitousFelicitousness Date: Thu, 9 Apr 2026 09:11:51 +0100 Subject: [PATCH] fix(lora): case-insensitive stage detection for two-stage LoRA names --- modules/lora/extra_networks_lora.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modules/lora/extra_networks_lora.py b/modules/lora/extra_networks_lora.py index 82e5e11df..364244b1a 100644 --- a/modules/lora/extra_networks_lora.py +++ b/modules/lora/extra_networks_lora.py @@ -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())