298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-20 03:13:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add TE support for Lumina 2 LoRA

Browse Source
This commit is contained in:
rockerBOO 2025-03-04 12:10:06 -05:00
parent cf0b549d48
commit eaf0826bd7
No known key found for this signature in database
GPG Key ID: 0D4EAF00DCABC97B
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/lora.py
View File

@ -268,6 +268,9 @@ def model_lora_keys_clip(model, key_map={}):
for k in sdk: for k in sdk:
if k.endswith(".weight"): 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 if k.startswith("t5xxl.transformer."):#OneTrainer SD3 and Flux lora
l_key = k[len("t5xxl.transformer."):-len(".weight")] l_key = k[len("t5xxl.transformer."):-len(".weight")]
t5_index = 1 t5_index = 1