Skip to content

Commit ab1050b

Browse files
Support ace step 1.5 base model loras. (Comfy-Org#12252)
1 parent fb23935 commit ab1050b

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

comfy/lora.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -332,6 +332,12 @@ def model_lora_keys_unet(model, key_map={}):
332332
key_map["{}".format(key_lora)] = k
333333
key_map["transformer.{}".format(key_lora)] = k
334334

335+
if isinstance(model, comfy.model_base.ACEStep15):
336+
for k in sdk:
337+
if k.startswith("diffusion_model.decoder.") and k.endswith(".weight"):
338+
key_lora = k[len("diffusion_model.decoder."):-len(".weight")]
339+
key_map["base_model.model.{}".format(key_lora)] = k # Official base model loras
340+
335341
return key_map
336342

337343

0 commit comments

Comments
 (0)