diff --git a/comfy/lora.py b/comfy/lora.py
index e54b3b4f..c43a6fe4 100644
--- a/comfy/lora.py
+++ b/comfy/lora.py
@@ -384,6 +384,7 @@ def model_lora_keys_unet(model, key_map={}):
                 key_lora = key_lora.replace(".modulation.lin.", ".modulation.linear.")
                 key_lora = key_lora[len("diffusion_model."):-len(".weight")]
                 key_map["transformer.{}".format(key_lora)] = k
+                key_map["diffusion_model.{}".format(key_lora)] = k  # Old loras
 
     return key_map