diff --git a/comfy/lora.py b/comfy/lora.py index 83e55ec0732..2d0ed7bf54b 100644 --- a/comfy/lora.py +++ b/comfy/lora.py @@ -294,6 +294,7 @@ def model_lora_keys_unet(model, key_map={}): unet_key = "diffusion_model.{}".format(diffusers_keys[k]) key_lora = k[:-len(".weight")].replace(".", "_") key_map["lora_unet_{}".format(key_lora)] = unet_key + key_map["lycoris_{}".format(key_lora)] = unet_key #simpletuner lycoris format diffusers_lora_prefix = ["", "unet."] for p in diffusers_lora_prefix: