Simpletrainer lycoris format.

MoonRide303 · Aug 20, 2024 · ea63b1c · ea63b1c
1 parent 9953f22
commit ea63b1c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/comfy/lora.py b/comfy/lora.py
@@ -318,7 +318,7 @@ def model_lora_keys_unet(model, key_map={}):
         for k in diffusers_keys:
             if k.endswith(".weight"):
                 to = diffusers_keys[k]
-                key_lora = "transformer.{}".format(k[:-len(".weight")]) #simpletrainer and probably regular diffusers flux lora format
-                key_map[key_lora] = to
+                key_map["transformer.{}".format(k[:-len(".weight")])] = to #simpletrainer and probably regular diffusers flux lora format
+                key_map["lycoris_{}".format(k[:-len(".weight")].replace(".", "_"))] = to #simpletrainer lycoris
 
     return key_map