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

update

Browse Source
This commit is contained in:
yushan777
2023-09-24 15:24:29 +01:00
parent a2c0f3644b
commit 51e1b45abd
8 changed files with 8 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
train_db.py
View File

@@ -361,9 +361,7 @@ def train(args):
)
# 指定ステップごとにモデルを保存
if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and \
args.save_every_n_steps_after_x is not None and global_step >= args.save_every_n_steps_after_x:
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:
src_path = src_stable_diffusion_ckpt if save_stable_diffusion_format else src_diffusers_model_path