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

fix typos

Browse Source
This commit is contained in:
Kohya S
2023-04-13 21:18:18 +09:00
parent a8632b7329
commit 2de9a51591
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
networks/extract_lora_from_dylora.py
View File

@@ -43,7 +43,7 @@ def split_lora_model(lora_sd, unit):
print(f"Max rank: {max_rank}")
rank = unit
splitted_models = []
split_models = []
new_alpha = None
while rank < max_rank:
print(f"Splitting rank {rank}")
@@ -62,10 +62,10 @@ def split_lora_model(lora_sd, unit):
# new_sd[key] = new_alpha
new_sd[key] = value
splitted_models.append((new_sd, rank, new_alpha))
split_models.append((new_sd, rank, new_alpha))
rank += unit
return max_rank, splitted_models
return max_rank, split_models
def split(args):
@@ -73,10 +73,10 @@ def split(args):
lora_sd, metadata = load_state_dict(args.model)
print("Splitting Model...")
original_rank, splitted_models = split_lora_model(lora_sd, args.unit)
original_rank, split_models = split_lora_model(lora_sd, args.unit)
comment = metadata.get("ss_training_comment", "")
for state_dict, new_rank, new_alpha in splitted_models:
for state_dict, new_rank, new_alpha in split_models:
# update metadata
if metadata is None:
new_metadata = {}