From bb977f669326294da487db33b118e428cb3d8463 Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Tue, 15 Apr 2025 01:55:11 +0800
Subject: [PATCH] Update train_util.py

---
 library/train_util.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/library/train_util.py b/library/train_util.py
index 141b1b46..7c69fcb2 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -5595,7 +5595,6 @@ def sample_images_common(
     with torch.no_grad():
         with distributed_state.split_between_processes(prompts) as prompt_dict_lists:
             for prompt_dict in prompt_dict_lists:
-                if '__caption__' in prompt_dict.get("prompt"):
                 sample_image_inference(
                     accelerator, args, pipeline, save_dir, prompt_dict, epoch, steps, prompt_replacement, controlnet=controlnet
                 )