github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-01-20 05:18:17 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix bug in FLUX T5 Koyha-style LoRA key parsing.

Browse Source
This commit is contained in:
Ryan Dick
2025-01-23 17:35:30 +00:00
parent d9c1c7d63d
commit eeff9d3df5
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/backend/patches/lora_conversions/flux_kohya_lora_conversion_utils.py
View File

@@ -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():