Fix some new loras.

2 years ago · 727588d076
parent 315ba30c81
commit 727588d076
1 changed files with 2 additions and 0 deletions
--- a/comfy/sd.py
+++ b/comfy/sd.py
@ -170,6 +170,8 @@ def model_lora_keys_clip(model, key_map={}):
            if k in sdk:
                lora_key = text_model_lora_key.format(b, LORA_CLIP_MAP[c])
                key_map[lora_key] = k
+                lora_key = "lora_te1_text_model_encoder_layers_{}_{}".format(b, LORA_CLIP_MAP[c])
+                key_map[lora_key] = k

            k = "clip_l.transformer.text_model.encoder.layers.{}.{}.weight".format(b, c)
            if k in sdk: