This commit is contained in:
Dave Lage 2026-07-17 23:28:17 -07:00 committed by GitHub
commit 48a0826f92
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions

View File

@ -147,6 +147,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