Update train_util.py

2026-04-15 16:39:42 +00:00 · 2025-04-15 01:55:11 +08:00
parent 8ce9dca025
commit bb977f6693
1 changed files with 0 additions and 1 deletions
--- 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
                )