From 19e6bf0979d624bfe3aae2c24b89892ca7c8ecc9 Mon Sep 17 00:00:00 2001 From: DKnight54 <126916963+DKnight54@users.noreply.github.com> Date: Sat, 8 Feb 2025 21:47:02 +0800 Subject: [PATCH] Update accel_sdxl_gen_img.py --- accel_sdxl_gen_img.py | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py index 8e976469..35d99c63 100644 --- a/accel_sdxl_gen_img.py +++ b/accel_sdxl_gen_img.py @@ -2939,22 +2939,22 @@ def main(args): distributed_state.wait_for_everyone() if distributed_state.is_main_process: batchlogstr = "Running through ext_separated_list_of_batches Before Gather:\n" - for x in range(len(ext_separated_list_of_batches)): - batchlogstr += f"Batch_ext {x} of {len(ext_separated_list_of_batches)} break:\n" - for y in range(len(ext_separated_list_of_batches[x])): - batchlogstr += f" Batch {y} of {len(ext_separated_list_of_batches[x])} break:\n" - for z in range(len(ext_separated_list_of_batches[x][y])): - batchlogstr += f" Image {z} of {len(ext_separated_list_of_batches[x][y])} break: {ext_separated_list_of_batches[x][y][z].global_count}\n" + for x in range(len(ext_separated_list_of_batches)): + batchlogstr += f"Batch_ext {x} of {len(ext_separated_list_of_batches)} break:\n" + for y in range(len(ext_separated_list_of_batches[x])): + batchlogstr += f" Batch {y} of {len(ext_separated_list_of_batches[x])} break:\n" + for z in range(len(ext_separated_list_of_batches[x][y])): + batchlogstr += f" Image {z} of {len(ext_separated_list_of_batches[x][y])} break: {ext_separated_list_of_batches[x][y][z].global_count}\n" logger.info(batchlogstr) ext_separated_list_of_batches = gather_object(ext_separated_list_of_batches) if distributed_state.is_main_process: batchlogstr = "Running through ext_separated_list_of_batches After Gather:\n" - for x in range(len(ext_separated_list_of_batches)): - batchlogstr += f"Batch_ext {x} of {len(ext_separated_list_of_batches)} break:\n" - for y in range(len(ext_separated_list_of_batches[x])): - batchlogstr += f" Batch {y} of {len(ext_separated_list_of_batches[x])} break:\n" - for z in range(len(ext_separated_list_of_batches[x][y])): - batchlogstr += f" Image {z} of {len(ext_separated_list_of_batches[x][y])} break: {ext_separated_list_of_batches[x][y][z].global_count}\n" + for x in range(len(ext_separated_list_of_batches)): + batchlogstr += f"Batch_ext {x} of {len(ext_separated_list_of_batches)} break:\n" + for y in range(len(ext_separated_list_of_batches[x])): + batchlogstr += f" Batch {y} of {len(ext_separated_list_of_batches[x])} break:\n" + for z in range(len(ext_separated_list_of_batches[x][y])): + batchlogstr += f" Image {z} of {len(ext_separated_list_of_batches[x][y])} break: {ext_separated_list_of_batches[x][y][z].global_count}\n" logger.info(batchlogstr) del extinfo #logger.info(f"\nDevice {distributed_state.device}: {ext_separated_list_of_batches}")