From 79962b8e8e8cd31a790a38560b0c8a2af02d4144 Mon Sep 17 00:00:00 2001 From: DKnight54 <126916963+DKnight54@users.noreply.github.com> Date: Sat, 8 Feb 2025 18:39:10 +0800 Subject: [PATCH] Update accel_sdxl_gen_img.py --- accel_sdxl_gen_img.py | 1 + 1 file changed, 1 insertion(+) diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py index 6d28b945..22cf9fa6 100644 --- a/accel_sdxl_gen_img.py +++ b/accel_sdxl_gen_img.py @@ -2913,6 +2913,7 @@ def main(args): sublist.extend(split_into_batches.pop(-1)) split_into_batches = [] + sublist.reverse() n, m = divmod(len(sublist), distributed_state.num_processes) split_into_batches.extend([sublist[i*n+min(i,m):(i+1)*n+min(i+1,m)] for i in range(distributed_state.num_processes)]) ext_separated_list_of_batches.append(split_into_batches)