diff --git a/comfy/lora.py b/comfy/lora.py index ef110c164..387d5c52a 100644 --- a/comfy/lora.py +++ b/comfy/lora.py @@ -283,8 +283,9 @@ def model_lora_keys_unet(model, key_map={}): for k in sdk: if k.startswith("diffusion_model."): if k.endswith(".weight"): - key_lora = k[len("diffusion_model."):-len(".weight")].replace(".", "_") - key_map["lycoris_{}".format(key_lora)] = k #SimpleTuner lycoris format + key_lora = k[len("diffusion_model."):-len(".weight")] + 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.ACEStep): for k in sdk: