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

fix: don't save state if no --save-state arg given

Browse Source
This commit is contained in:
Evgeny Chukreev
2023-05-18 20:09:06 +02:00
parent c924c47f37
commit 0c942106bf
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
library/train_util.py
View File

@@ -3167,10 +3167,11 @@ def save_sd_model_on_epoch_end_or_stepwise(
print(f"removing old model: {remove_out_dir}") print(f"removing old model: {remove_out_dir}")
shutil.rmtree(remove_out_dir) shutil.rmtree(remove_out_dir)
if on_epoch_end: if args.save_state:
save_and_remove_state_on_epoch_end(args, accelerator, epoch_no) if on_epoch_end:
else: save_and_remove_state_on_epoch_end(args, accelerator, epoch_no)
save_and_remove_state_stepwise(args, accelerator, global_step) else:
save_and_remove_state_stepwise(args, accelerator, global_step)
def save_and_remove_state_on_epoch_end(args: argparse.Namespace, accelerator, epoch_no): def save_and_remove_state_on_epoch_end(args: argparse.Namespace, accelerator, epoch_no):