wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-20 05:18:15 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Fix: HiDream-O1 LoRA prefix for landing

Browse Source
This commit is contained in:
yjy415 2026-07-13 19:26:10 +08:00
parent bd9086426f
commit c4f414081c
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/lora.py
View File

@ -309,8 +309,6 @@ def model_lora_keys_unet(model, key_map={}):
if k.startswith("diffusion_model."): if k.startswith("diffusion_model."):
if k.endswith(".weight"): if k.endswith(".weight"):
key_lora = k[len("diffusion_model."):-len(".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 key_map["model.{}".format(key_lora)] = k
if isinstance(model, comfy.model_base.ACEStep): if isinstance(model, comfy.model_base.ACEStep):