From 77ca8713933a5349c03dcab7a9b0ed7063ef1e01 Mon Sep 17 00:00:00 2001 From: DKnight54 <126916963+DKnight54@users.noreply.github.com> Date: Sat, 25 Jan 2025 04:08:42 +0800 Subject: [PATCH] Update accel_sdxl_gen_img.py --- accel_sdxl_gen_img.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py index 245c656d..f378a457 100644 --- a/accel_sdxl_gen_img.py +++ b/accel_sdxl_gen_img.py @@ -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)):