From bd9086426f92d8964d3d3f6e661a3e4552976eac Mon Sep 17 00:00:00 2001 From: yjy415 <2471352175@qq.com> Date: Sat, 4 Jul 2026 20:59:23 +0800 Subject: [PATCH] Fix: HiDream-O1 LoRA loading --- comfy/lora.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/comfy/lora.py b/comfy/lora.py index 427cf98aa4f6..d29816830363 100644 --- 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