From 5a01f0695df9c7d73049ced1ac413632667e7bf9 Mon Sep 17 00:00:00 2001 From: DKnight54 <126916963+DKnight54@users.noreply.github.com> Date: Sat, 8 Feb 2025 23:33:49 +0800 Subject: [PATCH] Update accel_sdxl_gen_img.py --- accel_sdxl_gen_img.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py index 1d475d35..b5f9c10b 100644 --- a/accel_sdxl_gen_img.py +++ b/accel_sdxl_gen_img.py @@ -2972,12 +2972,9 @@ def main(args): if distributed_state.is_main_process: for image, metadata, filename in zip(all_images, all_metadatas, all_filenames): logger.info(f"Saving image: {fln}") - image.save(os.path.join(args.outdir, fln), pnginfo=metadata) + image.save(os.path.join(args.outdir, filename), pnginfo=metadata) distributed_state.wait_for_everyone() - - - distributed_state.wait_for_everyone() #for i in range(len(data_loader)): # logger.info(f"Loading Batch {i+1} of {len(data_loader)}") # prompt_data_list_split.append(data_loader[i])