Add simpletuner lycoris format for SD unet.

comfyanonymous · Sep 30, 2024 · b4626ab · b4626ab
1 parent a9e459c
commit b4626ab
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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: