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

Add training metadata to output LoRA model

Browse Source
This commit is contained in:
space-nuko
2023-01-10 02:49:52 -08:00
parent a84ca297bd
commit 2e4ce0fdff
4 changed files with 72 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
networks/extract_lora_from_models.py
View File

@@ -135,7 +135,7 @@ def svd(args):
if dir_name and not os.path.exists(dir_name):
os.makedirs(dir_name, exist_ok=True)
lora_network_o.save_weights(args.save_to, save_dtype)
lora_network_o.save_weights(args.save_to, save_dtype, {})
print(f"LoRA weights are saved to: {args.save_to}")