From 76750eec05aa2281b2f79bb409ca99eddf227008 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 25 Jul 2025 17:24:43 +0300
Subject: [PATCH] Allow loading aitoolkit/lycoris format

---
 nodes_model_loading.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/nodes_model_loading.py b/nodes_model_loading.py
index 0b0a81f..9157d0e 100644
--- a/nodes_model_loading.py
+++ b/nodes_model_loading.py
@@ -106,6 +106,13 @@ def filter_state_dict_by_blocks(state_dict, blocks_mapping, layer_filter=[]):
 def standardize_lora_key_format(lora_sd):
     new_sd = {}
     for k, v in lora_sd.items():
+        # aitoolkit/lycoris format
+        k = k.replace("lycoris_blocks_", "blocks.")
+        k = k.replace("_cross_attn_", ".cross_attn.")
+        k = k.replace("_self_attn_", ".self_attn.")
+        k = k.replace("_ffn_net_0_proj", ".ffn.0")
+        k = k.replace("_ffn_net_2", ".ffn.2")
+        k = k.replace("to_out_0", "o")
         # Diffusers format
         if k.startswith('transformer.'):
             k = k.replace('transformer.', 'diffusion_model.')