diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py index ea05fbf6..dea87ff2 100644 --- a/accel_sdxl_gen_img.py +++ b/accel_sdxl_gen_img.py @@ -2937,6 +2937,15 @@ def main(args): ext_separated_list_of_batches[x] = holder[:] ''' distributed_state.wait_for_everyone() + if distributed_state.is_main_process: + batchlogstr = "Running through ext_separated_list_of_batches:\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:\n" + logger.info(batchlogstr) ext_separated_list_of_batches = gather_object(ext_separated_list_of_batches) del extinfo #logger.info(f"\nDevice {distributed_state.device}: {ext_separated_list_of_batches}")