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

Update accel_sdxl_gen_img.py

Browse Source
This commit is contained in:
DKnight54
2025-01-25 04:08:42 +08:00
committed by GitHub
parent b927a7d345
commit 77ca871393
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
accel_sdxl_gen_img.py
View File

@@ -2854,10 +2854,12 @@ def main(args):
batch_index.append(batch_data[i])
test_batch_index.append(batch_data[i])
if (i+1) % args.batch_size == 0:
logger.info(f"Loading {batch_index}")
batch_data_split.append(batch_index)
batch_index.clear()
if (i+1) % 4 == 0:
test_batch_data_split.append(batch_index)
logger.info(f"Loading {test_batch_index}")
test_batch_data_split.append(test_batch_index)
test_batch_index.clear()
logger.info(f"batch_data_split: {batch_data_split}")
for i in range(len(test_batch_data_split)):