We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent fb23935 commit ab1050bCopy full SHA for ab1050b
comfy/lora.py
@@ -332,6 +332,12 @@ def model_lora_keys_unet(model, key_map={}):
332
key_map["{}".format(key_lora)] = k
333
key_map["transformer.{}".format(key_lora)] = k
334
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
+
341
return key_map
342
343
0 commit comments