From 9f47400cb02521d24daa3a58be92fb775b05e692 Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Sun, 26 Jan 2025 04:42:02 +0800
Subject: [PATCH] Update accel_sdxl_gen_img.py

---
 accel_sdxl_gen_img.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py
index 085f31bb..3d29cffe 100644
--- a/accel_sdxl_gen_img.py
+++ b/accel_sdxl_gen_img.py
@@ -2434,7 +2434,7 @@ def main(args):
         global_step = 0
         batch_data = []
         logger.info(f"args.interactive: {args.interactive}, (prompt_index: {prompt_index} < len(prompt_list): {len(prompt_list)} and not distributed_state.is_main_process: {not distributed_state.is_main_process})")
-        while args.interactive or (prompt_index < len(prompt_list) and not distributed_state.is_main_process):
+        while args.interactive or (prompt_index < len(prompt_list) and (not distributed_state.is_main_process or distributed_state.num_processes == 1)):
             if len(prompt_list) == 0:
                 # interactive
                 valid = False