diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py index 5de9b282..31b34eac 100644 --- a/accel_sdxl_gen_img.py +++ b/accel_sdxl_gen_img.py @@ -2822,7 +2822,7 @@ def main(args): 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 range(batch_list): + 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() @@ -2835,7 +2835,7 @@ 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 range(batch_list): + 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() @@ -2850,7 +2850,7 @@ 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 range(batch_list): + 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()