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

Save state when args.save_last_n_epochs_state is designated

Browse Source
This commit is contained in:
Yuta Hayashibe
2023-01-15 19:43:37 +09:00
parent a888223869
commit c6e28faa57
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/train_util.py
View File

@@ -1350,7 +1350,7 @@ def save_sd_model_on_epoch_end(args: argparse.Namespace, accelerator, src_path:
remove_old_func = remove_du
saving = save_on_epoch_end(args, save_func, remove_old_func, epoch_no, num_train_epochs)
if saving and args.save_state:
if saving and args.save_state or args.save_last_n_epochs_state is not None:
save_state_on_epoch_end(args, accelerator, model_name, epoch_no)