Update train_network.py

2026-04-15 08:36:41 +00:00 · 2025-02-03 05:09:14 +08:00
parent 07a2e3571f
commit e250e7aad0
1 changed files with 1 additions and 1 deletions
--- 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