Merge bd9086426f into b08debceca

Fix: HiDream-O1 LoRA loading
2026-07-21 13:58:15 +08:00 · 2026-07-06 09:06:42 +07:00 · 2026-07-04 20:59:23 +08:00
1 changed files with 9 additions and 0 deletions
--- a/comfy/lora.py
+++ b/comfy/lora.py
@ -304,6 +304,15 @@ def model_lora_keys_unet(model, key_map={}):
                    key_map["lycoris_{}".format(key_lora.replace(".", "_"))] = k #SimpleTuner lycoris format
                    key_map["transformer.{}".format(key_lora)] = k #SimpleTuner regular format

+    if isinstance(model, comfy.model_base.HiDreamO1):
+        for k in sdk:
+            if k.startswith("diffusion_model."):
+                if k.endswith(".weight"):
+                    key_lora = k[len("diffusion_model."):-len(".weight")]
+                    key_map["lycoris_{}".format(key_lora.replace(".", "_"))] = k
+                    key_map["transformer.{}".format(key_lora)] = k
+                    key_map["model.{}".format(key_lora)] = k
+
    if isinstance(model, comfy.model_base.ACEStep):
        for k in sdk:
            if k.startswith("diffusion_model.") and k.endswith(".weight"): #Official ACE step lora format
Author	SHA1	Message	Date
Jinyan Ye	ffb8467e57	Merge `bd9086426f` into `b08debceca`	2026-07-06 09:06:42 +07:00
yjy415	bd9086426f	Fix: HiDream-O1 LoRA loading	2026-07-04 20:59:23 +08:00