We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e4b0bb8 commit 06f85e2Copy full SHA for 06f85e2
1 file changed
comfy/lora.py
@@ -99,6 +99,9 @@ def model_lora_keys_clip(model, key_map={}):
99
for k in sdk:
100
if k.endswith(".weight"):
101
key_map["text_encoders.{}".format(k[:-len(".weight")])] = k #generic lora format without any weird key names
102
+ tp = k.find(".transformer.") #also map without wrapper prefix for composite text encoder models
103
+ if tp > 0 and not k.startswith("clip_"):
104
+ key_map["text_encoders.{}".format(k[tp + 1:-len(".weight")])] = k
105
106
text_model_lora_key = "lora_te_text_model_encoder_layers_{}_{}"
107
clip_l_present = False
0 commit comments