lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-12 00:19:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
layerdiffusion
2024-09-08 17:24:53 -07:00
parent c2c3decab2
commit f40930c55b
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
backend/patcher/lora.py
View File

@@ -31,11 +31,11 @@ def model_lora_keys_clip(model, key_map={}):
for model_key in model_keys:
if model_key.endswith(".weight"):
if model_key.startswith("t5xxl.transformer."):
# Flux OneTrainer T5
formatted = inner_str(model_key, "t5xxl.transformer.", ".weight")
formatted = formatted.replace(".", "_")
formatted = f"lora_te2_{formatted}"
key_map[formatted] = model_key
for prefix in ['te1', 'te2', 'te3']:
formatted = inner_str(model_key, "t5xxl.transformer.", ".weight")
formatted = formatted.replace(".", "_")
formatted = f"lora_{prefix}_{formatted}"
key_map[formatted] = model_key
return key_maps