From eaf4a9a16487868cefc2d019dc19742de4b92e2f Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Sat, 1 Feb 2025 23:17:03 +0800
Subject: [PATCH] Update accel_sdxl_gen_img.py

---
 accel_sdxl_gen_img.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py
index b30edc11..5ed67d2f 100644
--- a/accel_sdxl_gen_img.py
+++ b/accel_sdxl_gen_img.py
@@ -2900,6 +2900,7 @@ def main(args):
                 
                 for x in range(len(ext_separated_list_of_batches)):
                     temp_list = []
+                    logger.info(f"start: ext_separated_list_of_batches[x]: {len(ext_separated_list_of_batches[x])}")
                     for i in range(distributed_state.num_processes):
                         temp_list.append(ext_separated_list_of_batches[x][i :: distributed_state.num_processes])
                     ext_separated_list_of_batches[x] = []
@@ -2924,8 +2925,7 @@ def main(args):
             for batch_list in ext_separated_list_of_batches:
                 with distributed_state.split_between_processes(batch_list) as batches:
                     for j in range(len(batches)):
-                        logger.info(f"\nLoading batch {j+1}/{len(batches)} of {len(batches[j])} prompts onto device {distributed_state.local_process_index}:")
-                        logger.info(f"batch_list:")
+                        logger.info(f"\nLoading batch {j+1}/{len(batches)} of {len(batches[j])} prompts onto device {distributed_state.local_process_index}:\nbatch_list:")
                         for i in range(len(batches[j])):
                             logger.info(f"Image: {batches[j][i].global_count}\nDevice {distributed_state.device}: Prompt {i+1}: {batches[j][i].base.prompt}\nNegative Prompt: {batches[j][i].base.negative_prompt}\nSeed: {batches[j][i].base.seed}")
                         prev_image = process_batch(batch_list[j], highres_fix)[0]