Flux onetrainer lora.

siliconflow · Sep 2, 2024 · d043997 · d043997
1 parent f1c2301
commit d043997
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/comfy/lora.py b/comfy/lora.py
@@ -324,6 +324,7 @@ def model_lora_keys_unet(model, key_map={}):
                 to = diffusers_keys[k]
                 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
+                key_map["lora_transformer_{}".format(k[:-len(".weight")].replace(".", "_"))] = to #onetrainer
 
     return key_map