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

Merge 15d84bcfb4 into cadcd3169b

Browse Source
This commit is contained in:
v0xie
2025-06-22 20:53:16 +02:00
committed by GitHub
parent cadcd3169b 15d84bcfb4
commit 206633f676
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
train_network.py
View File

@@ -1036,6 +1036,8 @@ class NetworkTrainer:
self.sample_images(accelerator, args, None, global_step, accelerator.device, vae, tokenizer, text_encoder, unet)
clean_memory_on_device(accelerator.device)
# 指定ステップごとにモデルを保存
if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0:
accelerator.wait_for_everyone()
@@ -1092,6 +1094,8 @@ class NetworkTrainer:
self.sample_images(accelerator, args, epoch + 1, global_step, accelerator.device, vae, tokenizer, text_encoder, unet)
clean_memory_on_device(accelerator.device)
# end of epoch
# metadata["ss_epoch"] = str(num_train_epochs)