Added sample_images() for --sample_at_first

2026-04-09 06:45:09 +00:00 · 2023-10-29 22:08:42 +09:00
parent 5c150675bf
commit 2c731418ad
5 changed files with 36 additions and 1 deletions
--- a/train_network.py
+++ b/train_network.py
@@ -749,7 +749,9 @@ class NetworkTrainer:
            current_epoch.value = epoch + 1

            metadata["ss_epoch"] = str(epoch + 1)
-
+            
+            # For --sample_at_first
+            self.sample_images(accelerator, args, epoch, global_step, accelerator.device, vae, tokenizer, text_encoder, unet)
            network.on_epoch_start(text_encoder, unet)

            for step, batch in enumerate(train_dataloader):