From 4456065d29b6914a64a11c5ac466dcdc0385d4b4 Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Fri, 28 Mar 2025 19:29:59 +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 ffd1cbbf..4c7d3bf3 100644
--- a/train_network.py
+++ b/train_network.py
@@ -929,7 +929,7 @@ class NetworkTrainer:
                         if torch.any(torch.isnan(latents)):
                             accelerator.print("NaN found in latents, replacing with zeros")
                             latents = torch.nan_to_num(latents, 0, out=latents)
-                        logger.info(f"Latents Shape: {latents.shape}")
+                    logger.info(f"Latents Shape: {latents.shape}")
                     latents = latents * self.vae_scale_factor
 
                     # get multiplier for each sample