diff --git a/comfy/lora.py b/comfy/lora.py index 2ed0acb9d..8b5a534cc 100644 --- a/comfy/lora.py +++ b/comfy/lora.py @@ -142,6 +142,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