From e607dc45ff7a74d34218b04cd64885a136ff1f03 Mon Sep 17 00:00:00 2001 From: DKnight54 <126916963+DKnight54@users.noreply.github.com> Date: Sat, 1 Feb 2025 12:04:49 +0800 Subject: [PATCH] Update accel_sdxl_gen_img.py --- accel_sdxl_gen_img.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py index 8a9bcd74..91c2a9ba 100644 --- a/accel_sdxl_gen_img.py +++ b/accel_sdxl_gen_img.py @@ -2874,6 +2874,7 @@ def main(args): global_step += 1 prompt_index += 1 + batch_data = gather_object(batch_data) batch_separated_list = [] if distributed_state.is_main_process and len(batch_data) > 0: unique_extinfo = list(set(extinfo)) @@ -2908,7 +2909,7 @@ def main(args): logger.info(f"batch_separated_list: {len(batch_separated_list)}") distributed_state.wait_for_everyone() batch_data = gather_object(batch_separated_list) - logger.info(f"batch_data: {len(batch_data)}") + logger.info(f"batch_data line 2912: {len(batch_data)}") del extinfo if len(batch_data) > 0: