Fix some new loras.

727588d0 · comfyanonymous · 315ba30c · 727588d0
Commit 727588d0 authored Jul 25, 2023 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

comfy/sd.py comfy/sd.py +2 -0

No files found.
--- 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: