diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py index 35d99c63..27ed79aa 100644 --- a/accel_sdxl_gen_img.py +++ b/accel_sdxl_gen_img.py @@ -2945,16 +2945,16 @@ def main(args): 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) + 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" + + 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)}:\n" + for y in range(len(ext_separated_list_of_batches[x])): + batchlogstr += f" Batch {y} of {len(ext_separated_list_of_batches[x])}:\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])}: {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}")