From 4a744c375c8d597cfc5d3ce86087c25038086dbe Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Sat, 8 Feb 2025 21:57:01 +0800
Subject: [PATCH] Update accel_sdxl_gen_img.py

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

diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py
index 27ed79aa..8062bdbc 100644
--- a/accel_sdxl_gen_img.py
+++ b/accel_sdxl_gen_img.py
@@ -2896,12 +2896,14 @@ def main(args):
                 res = [j for j, val in enumerate(prompt_data_list) if val.ext == unique_extinfo[i]]
                 for index in res:
                     templist.append(prompt_data_list[index])
+                '''
                 if distributed_state.num_processes > 1:
                     resorted_list = []
                     for i in range(distributed_state.num_processes):
                         resorted_list.append(templist[i :: distributed_state.num_processes])
                     for list_of_prompts in resorted_list:
                         templist.extend(list_of_prompts)
+                '''
                 split_into_batches = get_batches(items=templist, batch_size=args.batch_size).copy()
                 
                 sublist = []