Update accel_sdxl_gen_img.py

This commit is contained in:
DKnight54
2025-02-08 22:06:37 +08:00
committed by GitHub
parent 4a744c375c
commit 0ee1083aec

View File

@@ -2921,7 +2921,6 @@ 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)])