From 198aea95a6bcc07a1affa1e5e0f471193f43bc14 Mon Sep 17 00:00:00 2001 From: DKnight54 <126916963+DKnight54@users.noreply.github.com> Date: Thu, 30 Jan 2025 11:18:08 +0800 Subject: [PATCH] Update train_util.py --- library/train_util.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/library/train_util.py b/library/train_util.py index 8c114297..17269b5d 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -5557,8 +5557,8 @@ def sample_images_common( with distributed_state.split_between_processes(per_process_prompts) as prompt_dict_lists: for prompt_dict in prompt_dict_lists[0]: if prompt_dict["prompt"] == '__caption__': - logger.info("No training prompts loaded, skipping '__caption__' prompt.") - continue + logger.info("No training prompts loaded, skipping '__caption__' prompt.") + continue sample_image_inference( accelerator, args, pipeline, save_dir, prompt_dict, epoch, steps, prompt_replacement, controlnet=controlnet )