From c1e387f9f9321b85c47e32f33f1351cb46ef679c Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Fri, 7 Feb 2025 01:18:43 +0800
Subject: [PATCH] Update train_network.py

---
 train_network.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train_network.py b/train_network.py
index 7105a1fa..e7c67eb4 100644
--- a/train_network.py
+++ b/train_network.py
@@ -1028,7 +1028,7 @@ class NetworkTrainer:
                 if accelerator.sync_gradients:
                     progress_bar.update(1)
                     global_step += 1
-                    if args.sample_every_n_steps is not None and steps % args.sample_every_n_steps != 0:
+                    if args.sample_every_n_steps is not None and global_step % args.sample_every_n_steps == 0:
                         example_tuple = (latents, batch["captions"])
                         self.sample_images(accelerator, args, None, global_step, accelerator.device, vae, tokenizer, text_encoder, unet, example_tuple)