1580158
/
sd-webui-lora-block-weight
mirror of https://github.com/hako-mikan/sd-webui-lora-block-weight.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' of https://github.com/hako-mikan/sd-webui-lora-block-weight

main
hako-mikan 2025-02-02 20:07:58 +09:00
parent 5f04606bc5 ac0f869332
commit 6999768a7b
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/lora_block_weight.py
View File

@ -1233,7 +1233,7 @@ def lbwf(after_applying_lora_patches, ms, lwei, elements, starts, flux):
lvs = [v for v in vals if v[1][0] in LORAS]
for v in lvs:
ratio, picked = ratiodealer(key.replace(".","_"), l, e, flux)
n_vals.append((ratio * m if s is None or s == 0 else 0, *v[1:]))
n_vals.append([ratio * m if s is None or s == 0 else 0, *v[1:]])
if not picked:
errormodules.append(key)
lora_patches[key] = n_vals
@ -1376,8 +1376,10 @@ def get_flux_blocks(key):
return "VAE"
if "t5xxl" in key:
return "T5"
if "text_encoders.clip" in key:
if "clip_" in key:
return "CLIP"
if "t5xxl" in key:
return "T5"
match = re.search(r'\_(\d+)\_', key)
if "double_blocks" in key: