VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'dev' into sd3

Browse Source
This commit is contained in:
Kohya S
2024-09-19 21:15:39 +09:00
parent 1286e00bb0 d7e14721e2
commit 706a48d50e
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
networks/svd_merge_lora.py
View File

@@ -184,18 +184,19 @@ def get_lbw_block_index(lora_name: str, is_sdxl: bool = False) -> int:
elif "mid_block_" in lora_name: elif "mid_block_" in lora_name:
block_idx = 1 + NUM_OF_BLOCKS # 1-based index, num blocks, mid block block_idx = 1 + NUM_OF_BLOCKS # 1-based index, num blocks, mid block
else: else:
# SDXL: some numbers are skipped
if lora_name.startswith("lora_unet_"): if lora_name.startswith("lora_unet_"):
name = lora_name[len("lora_unet_") :] name = lora_name[len("lora_unet_") :]
if name.startswith("time_embed_") or name.startswith("label_emb_"): # 1, No LoRA in sd-scripts if name.startswith("time_embed_") or name.startswith("label_emb_"): # 1, No LoRA in sd-scripts
block_idx = 1 block_idx = 1
elif name.startswith("input_blocks_"): # 1-8 to 2-9 elif name.startswith("input_blocks_"): # 1-8 to 2-9
block_idx = 1 + int(name.split("_")[2]) block_idx = 1 + int(name.split("_")[2])
elif name.startswith("middle_block_"): # 10 elif name.startswith("middle_block_"): # 13
block_idx = 10 block_idx = 13
elif name.startswith("output_blocks_"): # 0-8 to 11-19 elif name.startswith("output_blocks_"): # 0-8 to 14-22
block_idx = 11 + int(name.split("_")[2]) block_idx = 14 + int(name.split("_")[2])
elif name.startswith("out_"): # 20, No LoRA in sd-scripts elif name.startswith("out_"): # 23, No LoRA in sd-scripts
block_idx = 20 block_idx = 23
return block_idx return block_idx