From eaf0826bd786f67544193b35be736e10ff86b6d8 Mon Sep 17 00:00:00 2001
From: rockerBOO <rockerboo@gmail.com>
Date: Tue, 4 Mar 2025 12:10:06 -0500
Subject: [PATCH] Add TE support for Lumina 2 LoRA

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

diff --git a/comfy/lora.py b/comfy/lora.py
index bc9f3022a0b..c075556110f 100644
--- a/comfy/lora.py
+++ b/comfy/lora.py
@@ -268,6 +268,9 @@ def model_lora_keys_clip(model, key_map={}):
 
     for k in sdk:
         if k.endswith(".weight"):
+            if k.startswith("gemma2_2b.transformer.model"): #Lumina
+                l_key = k[len("gemma2_2b.transformer.model."):-len(".weight")]
+                key_map["lora_te_{}".format(l_key.replace(".", "_"))] = k
             if k.startswith("t5xxl.transformer."):#OneTrainer SD3 and Flux lora
                 l_key = k[len("t5xxl.transformer."):-len(".weight")]
                 t5_index = 1