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

Merge pull request #2207 from kohya-ss/fix-flux-extract-lora-metadata-failed

Browse Source 
fix: update metadata construction to include model_config for flux
This commit is contained in:
Kohya S.
2025-09-24 21:19:27 +09:00
committed by GitHub
parent 121853ca2a 4b79d73504
commit 4c197a538b
2 changed files with 23 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
networks/flux_extract_lora.py
View File

@@ -139,7 +139,9 @@ def svd(
if not no_metadata:
title = os.path.splitext(os.path.basename(save_to))[0]
sai_metadata = sai_model_spec.build_metadata(lora_sd, False, False, False, True, False, time.time(), title, flux="dev")
sai_metadata = sai_model_spec.build_metadata(
lora_sd, False, False, False, True, False, time.time(), title, model_config={"flux": "dev"}
)
metadata.update(sai_metadata)
save_to_file(save_to, lora_sd, metadata, save_dtype)

22
networks/flux_merge_lora.py
View File

@@ -619,7 +619,16 @@ def merge(args):
merged_from = sai_model_spec.build_merged_from([args.flux_model] + args.models)
title = os.path.splitext(os.path.basename(args.save_to))[0]
sai_metadata = sai_model_spec.build_metadata(
None, False, False, False, False, False, time.time(), title=title, merged_from=merged_from, flux="dev"
None,
False,
False,
False,
False,
False,
time.time(),
title=title,
merged_from=merged_from,
model_config={"flux": "dev"},
)
if flux_state_dict is not None and len(flux_state_dict) > 0:
@@ -647,7 +656,16 @@ def merge(args):
merged_from = sai_model_spec.build_merged_from(args.models)
title = os.path.splitext(os.path.basename(args.save_to))[0]
sai_metadata = sai_model_spec.build_metadata(
flux_state_dict, False, False, False, True, False, time.time(), title=title, merged_from=merged_from, flux="dev"
flux_state_dict,
False,
False,
False,
True,
False,
time.time(),
title=title,
merged_from=merged_from,
model_config={"flux": "dev"},
)
metadata.update(sai_metadata)