From e250e7aad052ee81f86e12d60082c04b728f9348 Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Mon, 3 Feb 2025 05:09:14 +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 1a6ba5eb..b498df92 100644
--- a/train_network.py
+++ b/train_network.py
@@ -1028,7 +1028,7 @@ class NetworkTrainer:
                 #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"]):
+                for idx in range(len(batch["captions"])):
                     latent_dict = {}
                     latent_dict["prompt"] = batch["captions"][idx]
                     latent_dict["height"] = latents.shape[2] * 8