From 0933ee90f43d86dd8b0363161b86a809fd366f23 Mon Sep 17 00:00:00 2001 From: DKnight54 <126916963+DKnight54@users.noreply.github.com> Date: Sat, 25 Jan 2025 00:29:44 +0800 Subject: [PATCH] Update accel_sdxl_gen_img.py --- accel_sdxl_gen_img.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py index 31b34eac..66ac7f9c 100644 --- a/accel_sdxl_gen_img.py +++ b/accel_sdxl_gen_img.py @@ -2850,8 +2850,8 @@ def main(args): with torch.no_grad(): with distributed_state.split_between_processes(batch_data_split) as batch_list: logger.info(f"Loading batch of {len(batch_list)} prompts onto device {distributed_state.device}:") - for i in len(batch_list): - logger.info(f"Prompt {i}: {batch_list[i].base.prompt}") + for i in len(batch_list): + logger.info(f"Prompt {i}: {batch_list[i].base.prompt}") prev_image = process_batch(batch_list, highres_fix)[0] accelerator.wait_for_everyone() batch_data.clear()