Merge eaf0826bd7 into 2f7d8159c3

Add TE support for Lumina 2 LoRA
2025-04-16 00:23:30 +00:00 · 2025-04-08 23:41:39 +01:00 · 2025-03-04 12:10:06 -05:00
1 changed files with 3 additions and 0 deletions
--- 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
Author	SHA1	Message	Date
Dave Lage	e9996079cf	Merge `eaf0826bd7` into `2f7d8159c3`	2025-04-08 23:41:39 +01:00
rockerBOO	eaf0826bd7	Add TE support for Lumina 2 LoRA	2025-03-04 12:10:06 -05:00