From 4c5ec694f77080e2e05987e42d12df7bbc882e0a Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Mon, 3 Feb 2025 05:22:00 +0800
Subject: [PATCH] Update train_network.py

---
 train_network.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/train_network.py b/train_network.py
index b498df92..3a2243e6 100644
--- a/train_network.py
+++ b/train_network.py
@@ -1024,9 +1024,9 @@ class NetworkTrainer:
 
                 # Checks if the accelerator has performed an optimization step behind the scenes
                 # Collecting latents and caption lists from all processes
-                logger.info(f"latents.size: {latents.size()} before gather on device {accelerator.state.local_process_index}")
-                #Converts batch of latents into list of dicts containing individual latents, height and width to merge across processes
-                #Allows for different latent sizes
+                
+                # Converts batch of latents into list of dicts containing individual latents, height and width to merge across processes
+                # Allows for different latent sizes
                 latents_list = []
                 for idx in range(len(batch["captions"])):
                     latent_dict = {}