From 8cc503f57039b4f97b2cefc679b569dc7b25c67b Mon Sep 17 00:00:00 2001 From: DKnight54 <126916963+DKnight54@users.noreply.github.com> Date: Sun, 9 Feb 2025 00:43:07 +0800 Subject: [PATCH] Update accel_sdxl_gen_img.py --- accel_sdxl_gen_img.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py index d3bedccd..3b555ee3 100644 --- a/accel_sdxl_gen_img.py +++ b/accel_sdxl_gen_img.py @@ -2965,15 +2965,16 @@ def main(args): for i in range(len(batches[j])): batchlogstr += f"\nImage: {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}" logger.info(batchlogstr) - prev_image, prev_metadata, prev_filename = process_batch(batch_list[j], distributed_state, highres_fix) + coll_image, coll_metadata, coll_filename = process_batch(batch_list[j], distributed_state, highres_fix) logger.info(f"prev_image: {len(prev_image)}") logger.info(f"prev_metadata: {len(prev_metadata)}") logger.info(f"prev_filename: {len(prev_filename)}") distributed_state.wait_for_everyone() - all_images = gather_object(prev_image) - all_metadatas = gather_object(prev_metadata) - all_filenames = gather_object(prev_filename) + all_images = gather_object(coll_image) + all_metadatas = gather_object(coll_metadata) + all_filenames = gather_object(coll_filename) + prev_image = allimages[0] if distributed_state.is_main_process: for image, metadata, filename in zip(all_images, all_metadatas, all_filenames): logger.info(f"Saving image: {fln}")