Merge pull request #521 from akshaal/fix/save_state

fix: don't save state if no --save-state arg given
2026-04-08 22:35:09 +00:00 · 2023-05-21 08:48:48 +09:00
parent c971d9319c 0c942106bf
commit bc909e8359
1 changed files with 5 additions and 4 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -3167,6 +3167,7 @@ def save_sd_model_on_epoch_end_or_stepwise(
                print(f"removing old model: {remove_out_dir}")
                shutil.rmtree(remove_out_dir)

+    if args.save_state:
        if on_epoch_end:
            save_and_remove_state_on_epoch_end(args, accelerator, epoch_no)
        else: