VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-17 09:18:00 +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-01 12:47:31 +08:00
committed by GitHub
parent 91986ec4a2
commit 37331268e1
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
accel_sdxl_gen_img.py
View File

@@ -2879,9 +2879,10 @@ def main(args):
if len(batch_data) > 0:
if distributed_state.is_main_process:
unique_extinfo = list(set(extinfo))
logger.info(f"batch_data line 2880: {len(batch_data)}")
logger.info(f"Device {distributed_state.device}, batch_data line 2880: {len(batch_data)}, len(unique_extinfo): {len(unique_extinfo)}")
# splits list of prompts into sublists where BatchDataExt ext is identical
for i in range(len(unique_extinfo)):
logger.info(f"Device {distributed_state.device}, batch_data line 2880: {len(batch_data)}, len(unique_extinfo): {i}")
templist = []
res = [j for j, val in enumerate(batch_data) if val.ext == unique_extinfo[i]]
for index in res: