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

Merge 51e1b45abd into 63711390a0

Browse Source
This commit is contained in:
yushan777
2025-09-30 10:00:38 +05:30
committed by GitHub
parent 63711390a0 51e1b45abd
commit 7c4f5f78fd
9 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
train_network.py
View File

@@ -1037,7 +1037,8 @@ class NetworkTrainer:
self.sample_images(accelerator, args, None, global_step, accelerator.device, vae, tokenizer, text_encoder, unet)
# 指定ステップごとにモデルを保存
if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0:
if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and global_step >= args.save_every_n_steps_after_x:
accelerator.wait_for_everyone()
if accelerator.is_main_process:
ckpt_name = train_util.get_step_ckpt_name(args, "." + args.save_model_as, global_step)