From e2d5cd558b1bea63c061efdbee07cb553ad22a26 Mon Sep 17 00:00:00 2001
From: AI-Casanova <54461896+AI-Casanova@users.noreply.github.com>
Date: Fri, 11 Oct 2024 21:33:08 -0500
Subject: [PATCH] Flux Lycoris Keymapping

Progress on #3405
---
 extensions-builtin/Lora/lora_convert.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/extensions-builtin/Lora/lora_convert.py b/extensions-builtin/Lora/lora_convert.py
index be2ef356c..b7df0f84e 100644
--- a/extensions-builtin/Lora/lora_convert.py
+++ b/extensions-builtin/Lora/lora_convert.py
@@ -174,6 +174,8 @@ class KeyConvert:
             map_key = map_keys[position - 1]
             if search_key.startswith(map_key):
                 key = key.replace(map_key, self.UNET_CONVERSION_MAP[map_key]).replace("oft", "lora") # pylint: disable=unsubscriptable-object
+        if "lycoris" in key and "transformer" in key:
+            key = key.replace("lycoris", "lora_transformer")
         sd_module = shared.sd_model.network_layer_mapping.get(key, None)
         if sd_module is None:
             sd_module = shared.sd_model.network_layer_mapping.get(key.replace("guidance", "timestep"), None)  # FLUX1 fix