Skip to content

Commit d043997

Browse files
Flux onetrainer lora.
1 parent f1c2301 commit d043997

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

comfy/lora.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -324,6 +324,7 @@ def model_lora_keys_unet(model, key_map={}):
324324
to = diffusers_keys[k]
325325
key_map["transformer.{}".format(k[:-len(".weight")])] = to #simpletrainer and probably regular diffusers flux lora format
326326
key_map["lycoris_{}".format(k[:-len(".weight")].replace(".", "_"))] = to #simpletrainer lycoris
327+
key_map["lora_transformer_{}".format(k[:-len(".weight")].replace(".", "_"))] = to #onetrainer
327328

328329
return key_map
329330

0 commit comments

Comments
 (0)