From eeff9d3df54932cc040f34e85704698f37212d7a Mon Sep 17 00:00:00 2001 From: Ryan Dick Date: Thu, 23 Jan 2025 17:35:30 +0000 Subject: [PATCH] Fix bug in FLUX T5 Koyha-style LoRA key parsing. --- .../lora_conversions/flux_kohya_lora_conversion_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/invokeai/backend/patches/lora_conversions/flux_kohya_lora_conversion_utils.py b/invokeai/backend/patches/lora_conversions/flux_kohya_lora_conversion_utils.py index 5866878b94..41e41dbb51 100644 --- a/invokeai/backend/patches/lora_conversions/flux_kohya_lora_conversion_utils.py +++ b/invokeai/backend/patches/lora_conversions/flux_kohya_lora_conversion_utils.py @@ -154,10 +154,10 @@ def _convert_flux_t5_kohya_state_dict_to_invoke_format(state_dict: Dict[str, T]) """ def replace_func(match: re.Match[str]) -> str: - s = f"{match.group(1)}.{match.group(2)}.{match.group(3)}.{match.group(4)}" + s = f"encoder.block.{match.group(1)}.layer.{match.group(2)}.{match.group(3)}.{match.group(4)}" if match.group(5): s += f".{match.group(5)}" - return "encoder.block." + s + return s converted_dict: dict[str, T] = {} for k, v in state_dict.items():