From 6533b172c1eb6e01164672a2e8f6f8d42d16f755 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Sat, 24 Feb 2024 23:50:46 -0500
Subject: [PATCH] Support text encoder text_projection in lora.

---
 comfy/lora.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/comfy/lora.py b/comfy/lora.py
index 82d57fe..715e084 100644
--- a/comfy/lora.py
+++ b/comfy/lora.py
@@ -200,6 +200,13 @@ def model_lora_keys_clip(model, key_map={}):
                     lora_key = "lora_prior_te_text_model_encoder_layers_{}_{}".format(b, LORA_CLIP_MAP[c]) #cascade lora: TODO put lora key prefix in the model config
                     key_map[lora_key] = k
 
+
+    k = "clip_g.text_projection"
+    if k in sdk:
+        key_map["lora_prior_te_text_projection"] = k #cascade lora
+        # key_map["text_encoder.text_projection"] = k #TODO: check if other lora have the text_projection too
+        # key_map["lora_te_text_projection"] = k
+
     return key_map
 
 def model_lora_keys_unet(model, key_map={}):