mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-07-18 20:38:15 +08:00
Fix: HiDream-O1 LoRA prefix for landing
This commit is contained in:
parent
bd9086426f
commit
c4f414081c
@ -309,8 +309,6 @@ def model_lora_keys_unet(model, key_map={}):
|
||||
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):
|
||||
|
||||
Loading…
Reference in New Issue
Block a user