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

Make not to save state when args.save_state is False

Browse Source
This commit is contained in:
Yuta Hayashibe
2023-01-18 01:31:38 +09:00
parent 3815b82bef
commit 3eb8fb1875
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

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