VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-16 00:49:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update accel_sdxl_gen_img.py

Browse Source
This commit is contained in:
DKnight54
2025-02-09 01:45:27 +08:00
committed by GitHub
parent d69619cb29
commit b640c39b35
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
accel_sdxl_gen_img.py
View File

@@ -2927,7 +2927,13 @@ def main(args):
elif len(split_into_batches) == 1 :
sublist.extend(split_into_batches.pop(-1))
split_into_batches = []
# sublist = sorted(sublist, key=lambda x: x.global_count)
sublist = sorted(sublist, key=lambda x: x.global_count)
resorted_list = []
for i in range(distributed_state.num_processes):
resorted_list.append(sublist[i :: distributed_state.num_processes])
sublist = []
for list_of_prompts in resorted_list:
sublist.extend(list_of_prompts)
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)])