From 7d04dcb1a0931e458181e3a3ffab51e34dad5e1b Mon Sep 17 00:00:00 2001 From: DKnight54 <126916963+DKnight54@users.noreply.github.com> Date: Sun, 26 Jan 2025 02:46:56 +0800 Subject: [PATCH] Update accel_sdxl_gen_img.py --- accel_sdxl_gen_img.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py index e7444758..67dd4493 100644 --- a/accel_sdxl_gen_img.py +++ b/accel_sdxl_gen_img.py @@ -2457,7 +2457,9 @@ def main(args): if distributed_state.is_main_process: raw_prompts = handle_dynamic_prompt_variants(raw_prompt, args.images_per_prompt) raw_prompts = gather_object(raw_prompts) - + for pi in range(distributed_state.num_processes): + if pi == distributed_state.local_process_index: + logger.info(f"Total raw prompts: {len(raw_prompts)} for {distributed_state.local_process_index}") if distributed_state.is_main_process: # repeat prompt for pi in range(args.images_per_prompt if len(raw_prompts) == 1 else len(raw_prompts)):