From af8cf79a2dbdd541493d49c4f26c60d60d85ceae Mon Sep 17 00:00:00 2001
From: PsychoLogicAu <psycho.logic.au@gmail.com>
Date: Thu, 24 Oct 2024 16:18:32 +1100
Subject: [PATCH] support SimpleTuner lycoris lora for SD3 (#5340)

---
 comfy/lora.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/comfy/lora.py b/comfy/lora.py
index 81cd1696..b745ca4d 100644
--- a/comfy/lora.py
+++ b/comfy/lora.py
@@ -317,6 +317,10 @@ def model_lora_keys_unet(model, key_map={}):
                 key_lora = "lora_transformer_{}".format(k[:-len(".weight")].replace(".", "_")) #OneTrainer lora
                 key_map[key_lora] = to
 
+                key_lora = "lycoris_{}".format(k[:-len(".weight")].replace(".", "_")) #simpletuner lycoris format
+                key_map[key_lora] = to
+
+
     if isinstance(model, comfy.model_base.AuraFlow): #Diffusers lora AuraFlow
         diffusers_keys = comfy.utils.auraflow_to_diffusers(model.model_config.unet_config, output_prefix="diffusion_model.")
         for k in diffusers_keys: