From f935d42d8ee399e57028d33e0142730d0c163a91 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Fri, 25 Apr 2025 03:11:14 -0400
Subject: [PATCH] Support SimpleTuner lycoris lora format for HiDream.

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

diff --git a/comfy/lora.py b/comfy/lora.py
index 8760a21f..fff524be 100644
--- a/comfy/lora.py
+++ b/comfy/lora.py
@@ -279,6 +279,13 @@ def model_lora_keys_unet(model, key_map={}):
                 key_map["transformer.{}".format(key_lora)] = k
                 key_map["diffusion_model.{}".format(key_lora)] = k  # Old loras
 
+    if isinstance(model, comfy.model_base.HiDream):
+        for k in sdk:
+            if k.startswith("diffusion_model."):
+                if k.endswith(".weight"):
+                    key_lora = k[len("diffusion_model."):-len(".weight")].replace(".", "_")
+                    key_map["lycoris_{}".format(key_lora)] = k #SimpleTuner lycoris format
+
     return key_map